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

Blacklist the new TFLite converter for broken internal tests.

Browse Source 
PiperOrigin-RevId: 284868541
Change-Id: If4340c4c48aa2e2d779bfd4b4f916758498293d4
This commit is contained in:
Yu-Cheng Ling 2019-12-10 15:58:18 -08:00 committed by TensorFlower Gardener
parent 89991d2d5f
commit b735c799fd
2 changed files with 26 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -53,7 +53,25 @@ from tensorflow.python.saved_model import saved_model
from tensorflow.python.training.training_util import write_graph from tensorflow.python.training.training_util import write_graph
class TestModels(test_util.TensorFlowTestCase): class LiteTest(test_util.TensorFlowTestCase):
"""Base class of all the tests in this module."""
def setUp(self):
# Some cases are broken when we enable the new converter by default.
# Explicitly disabling it for now.
# TODO(b/145763157): Investigate if these are real issues.
self._original_use_experimental_new_converter = (
lite._USE_EXPERIMENTAL_NEW_CONVERTER)
lite._USE_EXPERIMENTAL_NEW_CONVERTER = False
super(LiteTest, self).setUp()
def tearDown(self):
super(LiteTest, self).tearDown()
lite._USE_EXPERIMENTAL_NEW_CONVERTER = (
self._original_use_experimental_new_converter)
class TestModels(LiteTest):
def assertValidDebugInfo(self, debug_info): def assertValidDebugInfo(self, debug_info):
"""Verify the DebugInfo is valid.""" """Verify the DebugInfo is valid."""
@ -1202,7 +1220,7 @@ class FromSessionTest(TestModels, parameterized.TestCase):
self.assertIn(('add@' + six.ensure_str(func)), converter._debug_info.traces) self.assertIn(('add@' + six.ensure_str(func)), converter._debug_info.traces)
class FromFrozenGraphFile(test_util.TensorFlowTestCase): class FromFrozenGraphFile(LiteTest):
def testFloat(self): def testFloat(self):
with ops.Graph().as_default(): with ops.Graph().as_default():
@ -1388,7 +1406,7 @@ class FromFrozenGraphFile(test_util.TensorFlowTestCase):
self.assertTrue(not converter._debug_info) self.assertTrue(not converter._debug_info)
class FromFrozenGraphObjectDetection(test_util.TensorFlowTestCase): class FromFrozenGraphObjectDetection(LiteTest):
def _initObjectDetectionArgs(self): def _initObjectDetectionArgs(self):
# Initializes the arguments required for the object detection model. # Initializes the arguments required for the object detection model.
@ -2082,7 +2100,7 @@ class GrapplerTest(TestModels, parameterized.TestCase):
self.assertEqual('Placeholder', input_details[0]['name']) self.assertEqual('Placeholder', input_details[0]['name'])
self.assertEqual('Const', input_details[1]['name']) self.assertEqual('Const', input_details[1]['name'])
class ImportOpsUtilTest(test_util.TensorFlowTestCase): class ImportOpsUtilTest(LiteTest):
def testGetPotentiallySupportedOps(self): def testGetPotentiallySupportedOps(self):
self.assertIsNotNone(lite.get_potentially_supported_ops()) self.assertIsNotNone(lite.get_potentially_supported_ops())

4
tensorflow/lite/testing/model_coverage/model_coverage_lib.py
View File

@ -96,6 +96,10 @@ def _convert(converter, **kwargs):
converter.optimizations = [_lite.Optimize.DEFAULT] converter.optimizations = [_lite.Optimize.DEFAULT]
if kwargs.get("quantize_to_float16", False): if kwargs.get("quantize_to_float16", False):
converter.target_spec.supported_types = [constants.FLOAT16] converter.target_spec.supported_types = [constants.FLOAT16]
# Some cases are broken when we enable the new converter by default.
# Explicitly disabling it for now.
# TODO(b/145763444): Investigate if these are real issues.
converter.experimental_new_converter = False
return converter.convert() return converter.convert()