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

Extended test to the case when new converter is enabled. 

Change-Id: I83f20d025027ad1266f99f9d79932cab4f1a9ed5
This commit is contained in:
Elena Zhelezina 2020-06-11 23:05:40 +01:00
parent dcfc2175c7
commit d03b86fe44
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/lite/python/lite_test.py
View File

@ -891,7 +891,11 @@ class FromSessionTest(TestModels, parameterized.TestCase):
('UseTfliteBuiltinsInt16DisableMLIR', ('UseTfliteBuiltinsInt16DisableMLIR',
[lite.OpsSet.\ [lite.OpsSet.\
EXPERIMENTAL_TFLITE_BUILTINS_ACTIVATIONS_INT16_WEIGHTS_INT8], EXPERIMENTAL_TFLITE_BUILTINS_ACTIVATIONS_INT16_WEIGHTS_INT8],
False)) False),
('UseTfliteBuiltinsInt16EnableMLIR',
[lite.OpsSet.\
EXPERIMENTAL_TFLITE_BUILTINS_ACTIVATIONS_INT16_WEIGHTS_INT8],
True))
def testCalibrateAndQuantizeBuiltinInt(self, supported_ops, enable_mlir): def testCalibrateAndQuantizeBuiltinInt(self, supported_ops, enable_mlir):
with ops.Graph().as_default(): with ops.Graph().as_default():
inp, output, calibration_gen = self._getCalibrationQuantizeModel() inp, output, calibration_gen = self._getCalibrationQuantizeModel()