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

Disable Python test due to error executing EagerPyFunc op in TFRT graph mode.

Browse Source 
PiperOrigin-RevId: 358013479
Change-Id: I905e6f4cf41271a0ed3c247d2d84f0188935a1bb
This commit is contained in:
Chuanhao Zhuge 2021-02-17 12:55:54 -08:00 committed by TensorFlower Gardener
parent c2604121fd
commit 7d45aa8560
2 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/python/kernel_tests/py_func_test.py
View File

@ -597,6 +597,7 @@ class EagerPyFuncTest(PyFuncTestBase):
self.assertIsNone(ret) self.assertIsNone(ret)
@test_util.run_in_graph_and_eager_modes @test_util.run_in_graph_and_eager_modes
@test_util.disable_tfrt("b/180469928")
def testEagerPyFuncInDefun(self): def testEagerPyFuncInDefun(self):
with test_util.device(use_gpu=True): with test_util.device(use_gpu=True):
def wrapper(): def wrapper():

2
tensorflow/python/ops/numpy_ops/np_interop_test.py
View File

@ -22,6 +22,7 @@ import numpy as onp
import tensorflow.compat.v2 as tf import tensorflow.compat.v2 as tf
from tensorflow.python.framework import ops from tensorflow.python.framework import ops
from tensorflow.python.framework import test_util
from tensorflow.python.ops import numpy_ops as np from tensorflow.python.ops import numpy_ops as np
from tensorflow.python.ops.numpy_ops import np_math_ops from tensorflow.python.ops.numpy_ops import np_math_ops
@ -229,6 +230,7 @@ class InteropTest(tf.test.TestCase):
# self.assertIsInstance(reduced, np.ndarray) # self.assertIsInstance(reduced, np.ndarray)
self.assertAllClose(reduced, 15) self.assertAllClose(reduced, 15)
@test_util.disable_tfrt('b/180469928')
def testPyFuncInterop(self): def testPyFuncInterop(self):
def py_func_fn(a, b): def py_func_fn(a, b):
return a + b return a + b