diff options
author | 2018-10-04 14:45:32 -0700 | |
---|---|---|
committer | 2018-10-04 14:49:07 -0700 | |
commit | ac7b84de8803edbb2d4da573b3f8704e9fad8fa8 (patch) | |
tree | 7775dd3b508b81d83955c362ae5a443d1c78b6c8 /tensorflow/contrib | |
parent | b74c9aa65fcbe615495a972a5021e983707d02f6 (diff) |
Internal change.
PiperOrigin-RevId: 215806953
Diffstat (limited to 'tensorflow/contrib')
-rw-r--r-- | tensorflow/contrib/lite/kernels/bidirectional_sequence_rnn.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/kernels/bidirectional_sequence_rnn.cc b/tensorflow/contrib/lite/kernels/bidirectional_sequence_rnn.cc index 9f62ac3f2c..c22a457a71 100644 --- a/tensorflow/contrib/lite/kernels/bidirectional_sequence_rnn.cc +++ b/tensorflow/contrib/lite/kernels/bidirectional_sequence_rnn.cc @@ -113,6 +113,7 @@ TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { // input configuration. TF_LITE_ENSURE_EQ(context, input->type, kTfLiteFloat32); + TF_LITE_ENSURE_EQ(context, input->dims->size, 3); const int batch_size = input->dims->data[0]; const int max_time = input->dims->data[1]; const int fw_num_units = fw_input_weights->dims->data[0]; |