From 95559baa01116996a0e6d657abaf31d8be505a22 Mon Sep 17 00:00:00 2001 From: Yu-Cheng Ling Date: Thu, 3 Oct 2019 21:15:53 -0700 Subject: [PATCH] Remove unnecessary reshape in lite_mlir_test PiperOrigin-RevId: 272803100 --- tensorflow/lite/python/lite_mlir_test.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/tensorflow/lite/python/lite_mlir_test.py b/tensorflow/lite/python/lite_mlir_test.py index cc2f4202037..1ac4e230d67 100644 --- a/tensorflow/lite/python/lite_mlir_test.py +++ b/tensorflow/lite/python/lite_mlir_test.py @@ -446,11 +446,8 @@ class FromConcreteFunctionTest(test_util.TensorFlowTestCase): # Check values from converted model. expected_value = concrete_func(input_data) - # TODO(haoliang): Remove this `reshape` op since it's not necessary. - actual_value = np.reshape( - self._evaluateTFLiteModel(tflite_model, [input_data]), (10, 10)) - for expected, actual in zip(expected_value, actual_value): - np.testing.assert_almost_equal(expected, actual) + actual_value = self._evaluateTFLiteModel(tflite_model, [input_data])[0] + np.testing.assert_almost_equal(expected_value, actual_value) class TestFlexMode(test_util.TensorFlowTestCase):