diff options
author | Alan Chiao <alanchiao@google.com> | 2018-07-12 12:16:09 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-12 12:25:21 -0700 |
commit | e8a65666c6aadbbbd2b19b9322d841b1547dbd35 (patch) | |
tree | eadeba71d8ad2d4c0c147d59f563d3246a7f6a65 /tensorflow/contrib/lite/kernels/lstm.cc | |
parent | 746a51b76742574d81783a4efe437e1824073d88 (diff) |
LSTM CHECK_OK on input tensor checks.
PiperOrigin-RevId: 204341675
Diffstat (limited to 'tensorflow/contrib/lite/kernels/lstm.cc')
-rw-r--r-- | tensorflow/contrib/lite/kernels/lstm.cc | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/contrib/lite/kernels/lstm.cc b/tensorflow/contrib/lite/kernels/lstm.cc index 3577ae6caa..4dfc891548 100644 --- a/tensorflow/contrib/lite/kernels/lstm.cc +++ b/tensorflow/contrib/lite/kernels/lstm.cc @@ -306,7 +306,8 @@ TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { const int n_output = recurrent_to_output_weights->dims->data[1]; // Check that input tensor dimensions matches with each other. - CheckInputTensorDimensions(context, node, n_input, n_output, n_cell); + TF_LITE_ENSURE_OK(context, CheckInputTensorDimensions(context, node, n_input, + n_output, n_cell)); // Get the pointer to output, activation_state and cell_state tensors. TfLiteTensor* output = GetOutput(context, node, kOutputTensor); |