Enable //third_party/tensorflow/python/kernel_tests:variables_test_tfrt test
PiperOrigin-RevId: 315977755 Change-Id: If1cf01e367fd2fe7f04ded600c730932a48baee7
This commit is contained in:
parent
d0dfb9b0bd
commit
66f4af41a0
tensorflow/python/kernel_tests
@ -1440,6 +1440,7 @@ tf_py_test(
|
||||
size = "small",
|
||||
srcs = ["variables_test.py"],
|
||||
tags = ["no_windows"], # b/133869052
|
||||
tfrt_enabled = True,
|
||||
deps = [
|
||||
"//tensorflow/python:array_ops",
|
||||
"//tensorflow/python:client_testlib",
|
||||
|
@ -174,6 +174,7 @@ class VariablesTestCase(test.TestCase, parameterized.TestCase):
|
||||
"Shapes.*and.*are incompatible"):
|
||||
var.assign(np.zeros(shape=[2, 2]))
|
||||
|
||||
@test_util.disable_tfrt("Graph is not supported yet. b/156187905")
|
||||
@test_util.run_in_graph_and_eager_modes
|
||||
def testAssignDifferentShapesAllowed(self):
|
||||
var = variables.Variable(np.zeros(shape=[1, 1]),
|
||||
@ -183,6 +184,8 @@ class VariablesTestCase(test.TestCase, parameterized.TestCase):
|
||||
self.evaluate(var.assign(np.zeros(shape=[2, 2])))
|
||||
self.assertAllEqual(np.zeros(shape=[2, 2]), var.read_value())
|
||||
|
||||
@test_util.disable_tfrt("GetHostSize() is not expected to be called with "
|
||||
"string type. b/156761465")
|
||||
def testZeroSizeStringAssign(self):
|
||||
with self.cached_session() as sess:
|
||||
array = variables.VariableV1(
|
||||
|
Loading…
Reference in New Issue
Block a user