Enable //third_party/tensorflow/python/kernel_tests:variables_test_tfrt test

PiperOrigin-RevId: 315977755
Change-Id: If1cf01e367fd2fe7f04ded600c730932a48baee7
This commit is contained in:
Kibeom Kim 2020-06-11 14:25:41 -07:00 committed by TensorFlower Gardener
parent d0dfb9b0bd
commit 66f4af41a0
2 changed files with 4 additions and 0 deletions
tensorflow/python/kernel_tests

View File

@ -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",

View File

@ -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(