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

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

Browse Source 
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/python/kernel_tests/BUILD
View File

@ -1440,6 +1440,7 @@ tf_py_test(
size = "small", size = "small",
srcs = ["variables_test.py"], srcs = ["variables_test.py"],
tags = ["no_windows"], # b/133869052 tags = ["no_windows"], # b/133869052
tfrt_enabled = True,
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib", "//tensorflow/python:client_testlib",

3
tensorflow/python/kernel_tests/variables_test.py
View File

@ -174,6 +174,7 @@ class VariablesTestCase(test.TestCase, parameterized.TestCase):
"Shapes.*and.*are incompatible"): "Shapes.*and.*are incompatible"):
var.assign(np.zeros(shape=[2, 2])) 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 @test_util.run_in_graph_and_eager_modes
def testAssignDifferentShapesAllowed(self): def testAssignDifferentShapesAllowed(self):
var = variables.Variable(np.zeros(shape=[1, 1]), 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.evaluate(var.assign(np.zeros(shape=[2, 2])))
self.assertAllEqual(np.zeros(shape=[2, 2]), var.read_value()) 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): def testZeroSizeStringAssign(self):
with self.cached_session() as sess: with self.cached_session() as sess:
array = variables.VariableV1( array = variables.VariableV1(