diff options
author | 2018-06-19 12:35:44 -0700 | |
---|---|---|
committer | 2018-06-19 12:38:27 -0700 | |
commit | 5fab6df2788937bee1cce3a4e8f5b9d1db7497ec (patch) | |
tree | ba18594841593a0b2a3eda55c076ca78c7bf0d4e /tensorflow/contrib/lite/kernels/test_util.cc | |
parent | 8f19772410ec20010e9930f9765dbd3aaeb06111 (diff) |
Support Variable Tensor API in LSTM Full kernel.
TFLite LSTM now supports 5 inputs, 18 inputs and 20 inputs.
PiperOrigin-RevId: 201222516
Diffstat (limited to 'tensorflow/contrib/lite/kernels/test_util.cc')
-rw-r--r-- | tensorflow/contrib/lite/kernels/test_util.cc | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tensorflow/contrib/lite/kernels/test_util.cc b/tensorflow/contrib/lite/kernels/test_util.cc index d23ec201b4..9156917140 100644 --- a/tensorflow/contrib/lite/kernels/test_util.cc +++ b/tensorflow/contrib/lite/kernels/test_util.cc @@ -32,8 +32,8 @@ std::vector<Matcher<float>> ArrayFloatNear(const std::vector<float>& values, return matchers; } -int SingleOpModel::AddInput(const TensorData& t) { - int id = AddTensor<float>(t, {}); +int SingleOpModel::AddInput(const TensorData& t, bool is_variable) { + int id = AddTensor<float>(t, {}, is_variable); inputs_.push_back(id); return id; } @@ -120,6 +120,7 @@ void SingleOpModel::BuildInterpreter( CHECK(interpreter_->AllocateTensors() == kTfLiteOk) << "Cannot allocate tensors"; + interpreter_->ResetVariableTensorsToZero(); } void SingleOpModel::Invoke() { CHECK(interpreter_->Invoke() == kTfLiteOk); } |