From 5fab6df2788937bee1cce3a4e8f5b9d1db7497ec Mon Sep 17 00:00:00 2001 From: Yu-Cheng Ling Date: Tue, 19 Jun 2018 12:35:44 -0700 Subject: Support Variable Tensor API in LSTM Full kernel. TFLite LSTM now supports 5 inputs, 18 inputs and 20 inputs. PiperOrigin-RevId: 201222516 --- tensorflow/contrib/lite/kernels/lstm_test.cc | 8 ++++++++ 1 file changed, 8 insertions(+) (limited to 'tensorflow/contrib/lite/kernels/lstm_test.cc') diff --git a/tensorflow/contrib/lite/kernels/lstm_test.cc b/tensorflow/contrib/lite/kernels/lstm_test.cc index 6da29a4a92..3f5c44a63e 100644 --- a/tensorflow/contrib/lite/kernels/lstm_test.cc +++ b/tensorflow/contrib/lite/kernels/lstm_test.cc @@ -97,6 +97,12 @@ class LSTMOpModel : public SingleOpModel { projection_bias_ = AddNullInput(); } + // Adding the 2 input state tensors. + input_activation_state_ = + AddInput(TensorData{TensorType_FLOAT32, {n_output_ * n_batch_}}, true); + input_cell_state_ = + AddInput(TensorData{TensorType_FLOAT32, {n_cell_ * n_batch_}}, true); + output_state_ = AddOutput(TensorType_FLOAT32); cell_state_ = AddOutput(TensorType_FLOAT32); output_ = AddOutput(TensorType_FLOAT32); @@ -227,6 +233,8 @@ class LSTMOpModel : public SingleOpModel { int projection_weights_; int projection_bias_; + int input_activation_state_; + int input_cell_state_; int output_; int output_state_; -- cgit v1.2.3