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

Temporarily blacklist keras model_coverage_lib_tests from TF2 testing 

This target appears to have never been executed under TF2 conditions,
and the Keras-specific tests are now failing.

PiperOrigin-RevId: 312770146
Change-Id: Icfc2ac6c7c73dda1db2c29a0a022d2ea8ea9c0da
This commit is contained in:
Jared Duke 2020-05-21 17:16:30 -07:00 committed by TensorFlower Gardener
parent 8d021e4030
commit b8b2dd9609
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -31,6 +31,7 @@ from tensorflow.python.eager import def_function
from tensorflow.python.framework import constant_op from tensorflow.python.framework import constant_op
from tensorflow.python.framework import dtypes from tensorflow.python.framework import dtypes
from tensorflow.python.framework import ops from tensorflow.python.framework import ops
from tensorflow.python.framework import test_util
from tensorflow.python.ops import array_ops from tensorflow.python.ops import array_ops
from tensorflow.python.ops import math_ops from tensorflow.python.ops import math_ops
from tensorflow.python.platform import test from tensorflow.python.platform import test
@ -178,18 +179,21 @@ class EvaluateKerasModel(test.TestCase):
os.close(fd) os.close(fd)
return keras_file return keras_file
@test_util.run_v1_only('Keras test fails under v2, see b/157266669')
def testFloat(self): def testFloat(self):
model = self._getSingleInputKerasModel() model = self._getSingleInputKerasModel()
keras_file = self._saveKerasModel(model) keras_file = self._saveKerasModel(model)
model_coverage.test_keras_model(keras_file) model_coverage.test_keras_model(keras_file)
@test_util.run_v1_only('Keras test fails under v2, see b/157266669')
def testPostTrainingQuantize(self): def testPostTrainingQuantize(self):
model = self._getSingleInputKerasModel() model = self._getSingleInputKerasModel()
keras_file = self._saveKerasModel(model) keras_file = self._saveKerasModel(model)
model_coverage.test_keras_model(keras_file, post_training_quantize=True) model_coverage.test_keras_model(keras_file, post_training_quantize=True)
@test_util.run_v1_only('Keras test fails under v2, see b/157266669')
def testTargetOps(self): def testTargetOps(self):
model = self._getSingleInputKerasModel() model = self._getSingleInputKerasModel()
keras_file = self._saveKerasModel(model) keras_file = self._saveKerasModel(model)