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

Enable tests blocked by b/117943489 

PiperOrigin-RevId: 311034341
Change-Id: Iab60ac2be4c1352aa39d3e266df09ba45a2feada
This commit is contained in:
Gaurav Jain 2020-05-11 18:10:26 -07:00 committed by TensorFlower Gardener
parent 9f7d843e3f
commit 75d9ed443a
3 changed files with 9 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/python/autograph/operators/data_structures_test.py
View File

@ -106,11 +106,12 @@ class ListTest(test.TestCase):
with self.cached_session() as sess: with self.cached_session() as sess:
self.assertAllEqual(self.evaluate(t), [[1, 2, 3]]) self.assertAllEqual(self.evaluate(t), [[1, 2, 3]])
@test_util.run_v1_only("b/117943489") @test_util.run_deprecated_v1
def test_append_tensorarray(self): def test_append_tensorarray(self):
l = tensor_array_ops.TensorArray(dtypes.int32, size=0, dynamic_size=True) l = tensor_array_ops.TensorArray(dtypes.int32, size=0, dynamic_size=True)
l1 = data_structures.list_append(l, 1) l1 = data_structures.list_append(l, 1)
l2 = data_structures.list_append(l1, 2) l2 = data_structures.list_append(l1, 2)
with self.cached_session() as sess: with self.cached_session() as sess:
self.assertAllEqual(self.evaluate(l1.stack()), [1]) self.assertAllEqual(self.evaluate(l1.stack()), [1])
self.assertAllEqual(self.evaluate(l2.stack()), [1, 2]) self.assertAllEqual(self.evaluate(l2.stack()), [1, 2])

1
tensorflow/python/autograph/utils/tensor_list_test.py
View File

@ -34,7 +34,6 @@ class TensorListTest(test.TestCase):
def _shape(self, shape_tuple): def _shape(self, shape_tuple):
return constant(shape_tuple, dtypes.int32) return constant(shape_tuple, dtypes.int32)
@test_util.run_v1_only("b/117943489")
def test_dynamic_list_append(self): def test_dynamic_list_append(self):
l = [] l = []
l = tl.dynamic_list_append(l, 1) l = tl.dynamic_list_append(l, 1)

15
tensorflow/python/kernel_tests/tensor_array_ops_test.py
View File

@ -1021,7 +1021,7 @@ class TensorArrayTest(test.TestCase):
# self._testWhileLoopWritePackGradients( # self._testWhileLoopWritePackGradients(
# dynamic_size=False, dtype=tf.int64) # dynamic_size=False, dtype=tf.int64)
@test_util.run_v1_only("b/117943489") @test_util.run_deprecated_v1
def testSkipEagerWhileLoopDynamicWritePackGradients(self): def testSkipEagerWhileLoopDynamicWritePackGradients(self):
self._testWhileLoopWritePackGradients( self._testWhileLoopWritePackGradients(
dynamic_size=True, dtype=dtypes.float32) dynamic_size=True, dtype=dtypes.float32)
@ -1251,7 +1251,6 @@ class TensorArrayTest(test.TestCase):
with self.assertRaises(ValueError): with self.assertRaises(ValueError):
w1.write(4, c2) w1.write(4, c2)
@test_util.run_v1_only("b/117943489")
def testUnpackShape(self): def testUnpackShape(self):
self._testUnpackShape() self._testUnpackShape()
@ -1340,11 +1339,11 @@ class TensorArrayTest(test.TestCase):
grad = gradients_impl.gradients(ys=[r], xs=[x]) grad = gradients_impl.gradients(ys=[r], xs=[x])
self.assertAllEqual(np.array([1.0, 1.0, 1.0]), self.evaluate(grad)[0]) self.assertAllEqual(np.array([1.0, 1.0, 1.0]), self.evaluate(grad)[0])
@test_util.run_v1_only("b/117943489") @test_util.run_deprecated_v1
def testSkipEagerTensorArrayUnpackDynamic(self): def testSkipEagerTensorArrayUnpackDynamic(self):
self._testTensorArrayUnpackDynamic() self._testTensorArrayUnpackDynamic()
@test_util.run_v1_only("b/117943489") @test_util.run_deprecated_v1
def testSkipEagerTensorArraySplitDynamic(self): def testSkipEagerTensorArraySplitDynamic(self):
with self.session(use_gpu=True) as sess: with self.session(use_gpu=True) as sess:
ta = tensor_array_ops.TensorArray( ta = tensor_array_ops.TensorArray(
@ -1422,7 +1421,7 @@ class TensorArrayTest(test.TestCase):
v2_msg if control_flow_util.ENABLE_CONTROL_FLOW_V2 else v1_msg): v2_msg if control_flow_util.ENABLE_CONTROL_FLOW_V2 else v1_msg):
ta.stack().eval() ta.stack().eval()
@test_util.run_v1_only("b/120545219") @test_util.run_deprecated_v1
def testSkipEagerTensorArrayEvalEmpty(self): def testSkipEagerTensorArrayEvalEmpty(self):
self._testTensorArrayEvalEmpty() self._testTensorArrayEvalEmpty()
@ -1445,11 +1444,11 @@ class TensorArrayTest(test.TestCase):
# first dimension of zero # first dimension of zero
self.assertAllEqual([0, 5], self.evaluate(concatenated).shape) self.assertAllEqual([0, 5], self.evaluate(concatenated).shape)
@test_util.run_v1_only("b/117943489") @test_util.run_deprecated_v1
def testSkipEagerTensorArrayEvalEmptyWithDefault(self): def testSkipEagerTensorArrayEvalEmptyWithDefault(self):
self._testTensorArrayEvalEmptyWithDefault() self._testTensorArrayEvalEmptyWithDefault()
@test_util.run_v1_only("b/117943489") @test_util.run_deprecated_v1
def testSkipEagerTensorArrayScatterReadAndGradients(self): def testSkipEagerTensorArrayScatterReadAndGradients(self):
with self.session(use_gpu=True) as session: with self.session(use_gpu=True) as session:
ta = tensor_array_ops.TensorArray( ta = tensor_array_ops.TensorArray(
@ -1476,7 +1475,7 @@ class TensorArrayTest(test.TestCase):
self.assertAllEqual([10.0, -10.0], read_vals[1]) self.assertAllEqual([10.0, -10.0], read_vals[1])
self.assertAllEqual([[2.0, 3.0], [4.0, 5.0]], grad_vals[0]) self.assertAllEqual([[2.0, 3.0], [4.0, 5.0]], grad_vals[0])
@test_util.run_v1_only("b/117943489") @test_util.run_deprecated_v1
def testSkipEagerTensorArrayScatterPartialReadAndGradients(self): def testSkipEagerTensorArrayScatterPartialReadAndGradients(self):
with self.session(use_gpu=True) as session: with self.session(use_gpu=True) as session:
ta = tensor_array_ops.TensorArray( ta = tensor_array_ops.TensorArray(