diff options
Diffstat (limited to 'src/tests/units/rnn_tests.cpp')
-rw-r--r-- | src/tests/units/rnn_tests.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/tests/units/rnn_tests.cpp b/src/tests/units/rnn_tests.cpp index 6405ef7a..5d67729b 100644 --- a/src/tests/units/rnn_tests.cpp +++ b/src/tests/units/rnn_tests.cpp @@ -181,7 +181,7 @@ void tests(DeviceType type, Type floatType = Type::float32) { auto context = concatenate({rnnFw.construct(graph)->transduce(input, mask), rnnBw.construct(graph)->transduce(input, mask)}, - /*axis =*/ input->shape().size() - 1); + /*axis =*/ (int)input->shape().size() - 1); if(second > 0) { // add more layers (unidirectional) by transducing the output of the |