diff options
author | Lev Proleev <levp@google.com> | 2020-01-14 15:40:39 +0000 |
---|---|---|
committer | Lev Proleev <levp@google.com> | 2020-02-05 12:42:29 +0000 |
commit | 42f879233ddad5e1511ebecb1826aab0647d79e8 (patch) | |
tree | 8ef5d50265b3845764c089e5c4f04e0cec8ce6ab /nn/runtime/test/TestValidateOperations.cpp | |
parent | f7898e6f63499b62322251f3982e22e13658266e (diff) | |
download | ml-42f879233ddad5e1511ebecb1826aab0647d79e8.tar.gz |
Add a state output for BIDIRECTIONAL_SEQUENCE_LSTM
Bug: 138443991
Test: NNTest_static and VtsHalNeuralnetworksV1_3TargetTest
Change-Id: I4f3292a2426093a6534567812dffb71e126f21f9
Diffstat (limited to 'nn/runtime/test/TestValidateOperations.cpp')
-rw-r--r-- | nn/runtime/test/TestValidateOperations.cpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/nn/runtime/test/TestValidateOperations.cpp b/nn/runtime/test/TestValidateOperations.cpp index e1f21539b..5c7e7789e 100644 --- a/nn/runtime/test/TestValidateOperations.cpp +++ b/nn/runtime/test/TestValidateOperations.cpp @@ -341,7 +341,7 @@ class OperationTestBase { return true; } std::vector<OperandTypeWithExtraParams> outputs = mValidOutputs; - for (int i = 0; i < 5; i++) { + for (int i = 0; i < 6; i++) { outputs.push_back(outputs[0]); if (ANEURALNETWORKS_NO_ERROR == addOperation(mValidInputs, outputs)) { if (mOpCode == ANEURALNETWORKS_UNIDIRECTIONAL_SEQUENCE_RNN && i < 1) { @@ -350,6 +350,9 @@ class OperationTestBase { if (mOpCode == ANEURALNETWORKS_UNIDIRECTIONAL_SEQUENCE_LSTM && i < 3) { continue; } + if (mOpCode == ANEURALNETWORKS_BIDIRECTIONAL_SEQUENCE_LSTM && i < 5) { + continue; + } return false; } } |