From 43758242b370999ae4f6b23a8c80f3e34443efe6 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Fri, 19 Apr 2019 11:36:45 -0700 Subject: [PATCH] Fix the check to accept Int8 format as well for Bidi-LSTM. PiperOrigin-RevId: 244393844 --- tensorflow/lite/kernels/bidirectional_sequence_lstm.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/tensorflow/lite/kernels/bidirectional_sequence_lstm.cc b/tensorflow/lite/kernels/bidirectional_sequence_lstm.cc index d5795ee3de0..7fc3666ea8f 100644 --- a/tensorflow/lite/kernels/bidirectional_sequence_lstm.cc +++ b/tensorflow/lite/kernels/bidirectional_sequence_lstm.cc @@ -185,6 +185,7 @@ TfLiteStatus CheckLstmTensorDimensionsAndTypes( TF_LITE_ENSURE_EQ(context, input_to_forget_weights->dims->data[0], n_cell); TF_LITE_ENSURE_EQ(context, input_to_forget_weights->dims->data[1], n_input); TF_LITE_ENSURE(context, (input_to_forget_weights->type == kTfLiteFloat32) || + (input_to_forget_weights->type == kTfLiteInt8) || (input_to_forget_weights->type == kTfLiteUInt8)); const TfLiteTensor* input_to_input_weights =