Extended test to the case when new converter is enabled.

Change-Id: I83f20d025027ad1266f99f9d79932cab4f1a9ed5
2020-06-11 23:05:40 +01:00 · 2020-06-11 23:05:40 +01:00 · d03b86fe44
parent dcfc2175c7
commit d03b86fe44
1 changed files with 5 additions and 1 deletions
--- a/tensorflow/lite/python/lite_test.py
+++ b/tensorflow/lite/python/lite_test.py
@ -891,7 +891,11 @@ class FromSessionTest(TestModels, parameterized.TestCase):
      ('UseTfliteBuiltinsInt16DisableMLIR',
       [lite.OpsSet.\
       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):
    with ops.Graph().as_default():
      inp, output, calibration_gen = self._getCalibrationQuantizeModel()