From d03b86fe4462ebad1f73d460c2aceab47372b239 Mon Sep 17 00:00:00 2001
From: Elena Zhelezina <elena.zhelezina@arm.com>
Date: Thu, 11 Jun 2020 23:05:40 +0100
Subject: [PATCH] Extended test to the case when new converter is enabled.

Change-Id: I83f20d025027ad1266f99f9d79932cab4f1a9ed5
---
 tensorflow/lite/python/lite_test.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/tensorflow/lite/python/lite_test.py b/tensorflow/lite/python/lite_test.py
index e6661c82894..478840c5549 100644
--- 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()