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

Mark tests that are not compatible with v2 behavior "run_v1_only".

Browse Source 
PiperOrigin-RevId: 345126240
Change-Id: I4e6c22b631b59db4712e515ed04542ae5e86a1fe
This commit is contained in:
Chuanhao Zhuge 2020-12-01 16:41:28 -08:00 committed by TensorFlower Gardener
parent 3bcd1d6337
commit 8493ce6116
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/python/framework/importer_test.py
View File

@ -32,6 +32,7 @@ from tensorflow.python.framework import importer
from tensorflow.python.framework import ops
from tensorflow.python.framework import tensor_shape
from tensorflow.python.framework import test_ops # pylint: disable=unused-import
from tensorflow.python.framework import test_util
from tensorflow.python.framework import versions
from tensorflow.python.ops import array_ops
from tensorflow.python.ops import control_flow_ops
@ -937,6 +938,7 @@ class ImportGraphDefTest(test.TestCase):
node { name: 'A' op: 'IntOutput' }
"""))
@test_util.run_v1_only("v1 Tensor doesn't have attribute 'numpy'")
def testWithExtensionAndAttr(self):
with ops.Graph().as_default() as g:
c = constant_op.constant(5.0, dtype=dtypes.float32, name="c")
@ -1219,6 +1221,8 @@ class ImportGraphDefTest(test.TestCase):
self.assertEqual(sess.run("external:0"), 11)
self.assertEqual(sess.run("outer:0"), 21)
@test_util.run_v1_only("import inside defun not supported when eager "
"execution is enabled.")
def testImportInsideDefun(self):
g = ops.Graph()
with g.as_default():
@ -1243,6 +1247,8 @@ class ImportGraphDefTest(test.TestCase):
z_val = self.evaluate(z)
self.assertEqual(z_val, -2.0)
@test_util.run_v1_only("_as_tf_output not supported when eager execution "
"is enabled.")
def testImportGraphWithFunctionTwice(self):
g = ops.Graph()
with g.as_default():