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

Fix //third_party/tensorflow/python:session_test. This failure is caused by constant folding.

Browse Source 
PiperOrigin-RevId: 351837260
Change-Id: Ia874803a85cc378a8fe0b6e89d9bbbcf928aa7b5
This commit is contained in:
Xiao Yu 2021-01-14 11:13:49 -08:00 committed by TensorFlower Gardener
parent 4133fef691
commit 7d43911faa
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tensorflow/python/client/session_test.py
View File

@ -1965,17 +1965,17 @@ class SessionTest(test_util.TensorFlowTestCase):
self.assertEqual(len(add_executions), 2)
@def_function.function
def fn():
a = constant_op.constant(1)
b = constant_op.constant(2)
def fn(a, b):
c = a + b
d = a + b
# These two AddV2 cannot use the same argument in tf.function since an
# optimization pass will remove duplicate ops and only run it once.
d = a + c
return c, d
with CaptureStderr() as log:
c, d = self.evaluate(fn())
c, d = self.evaluate(fn(constant_op.constant(1), constant_op.constant(2)))
self.assertEqual(c, 3)
self.assertEqual(d, 3)
self.assertEqual(d, 4)
# Ensure that we did log device placement.
add_executions = [l for l in str(log).splitlines() if 'AddV2' in l]
self.assertEqual(len(add_executions), 2)