experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Remove unnecessary reshape in lite_mlir_test

Browse Source 
PiperOrigin-RevId: 272803100
This commit is contained in:
Yu-Cheng Ling 2019-10-03 21:15:53 -07:00 committed by TensorFlower Gardener
parent 64f52623aa
commit 95559baa01
1 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/lite/python/lite_mlir_test.py
View File

@ -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):