From e8a65666c6aadbbbd2b19b9322d841b1547dbd35 Mon Sep 17 00:00:00 2001 From: Alan Chiao Date: Thu, 12 Jul 2018 12:16:09 -0700 Subject: LSTM CHECK_OK on input tensor checks. PiperOrigin-RevId: 204341675 --- tensorflow/contrib/lite/kernels/lstm_test.cc | 3 +++ 1 file changed, 3 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 0b7c56133e..0266f5fe57 100644 --- a/tensorflow/contrib/lite/kernels/lstm_test.cc +++ b/tensorflow/contrib/lite/kernels/lstm_test.cc @@ -13,6 +13,9 @@ See the License for the specific language governing permissions and limitations under the License. ==============================================================================*/ // Unit test for TFLite LSTM op. +// +// TODO(alanchiao): add unit test with invalid input dimensions for this and its +// variants. #include #include -- cgit v1.2.3