diff options
Diffstat (limited to 'nn/common/operations/RNN.cpp')
-rw-r--r-- | nn/common/operations/RNN.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/nn/common/operations/RNN.cpp b/nn/common/operations/RNN.cpp index dbff94f85..259c0915e 100644 --- a/nn/common/operations/RNN.cpp +++ b/nn/common/operations/RNN.cpp @@ -51,7 +51,7 @@ bool RNN::Prepare(const Operation& operation, RunTimeOperandInfo* operands, Shap NNTRACE_TRANS("RNN::Prepare"); // Check we have all the inputs and outputs we need. const int num_inputs = NumInputsWithValues(operation, operands); - NN_CHECK(num_inputs == 5 || num_inputs == 6); + NN_CHECK(num_inputs == 6); NN_CHECK_EQ(NumOutputs(operation), 2); const RunTimeOperandInfo* input = GetInput(operation, operands, kInputTensor); |