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

Remove @test_util.deprecated_graph_mode_only in dense_update_ops_test.py

Browse Source 
PiperOrigin-RevId: 324311282
Change-Id: I7f8c4fc89e4982dc885a8a07c155010d3f6097fd
This commit is contained in:
Kibeom Kim 2020-07-31 16:55:05 -07:00 committed by TensorFlower Gardener
parent 68bedc248d
commit 1b2ac91b25
1 changed files with 7 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -30,28 +30,28 @@ from tensorflow.python.platform import test
class AssignOpTest(test.TestCase):
def _initAssignFetch(self, x, y, use_gpu=False):
def _initAssignFetch(self, x, y, use_gpu):
"""Initialize a param to init and update it with y."""
super(AssignOpTest, self).setUp()
with self.cached_session(use_gpu=use_gpu):
with test_util.device(use_gpu=use_gpu):
p = variables.Variable(x)
assign = state_ops.assign(p, y)
self.evaluate(p.initializer)
new_value = self.evaluate(assign)
return self.evaluate(p), new_value
def _initAssignAddFetch(self, x, y, use_gpu=False):
def _initAssignAddFetch(self, x, y, use_gpu):
"""Initialize a param to init, and compute param += y."""
with self.cached_session(use_gpu=use_gpu):
with test_util.device(use_gpu=use_gpu):
p = variables.Variable(x)
add = state_ops.assign_add(p, y)
self.evaluate(p.initializer)
new_value = self.evaluate(add)
return self.evaluate(p), new_value
def _initAssignSubFetch(self, x, y, use_gpu=False):
def _initAssignSubFetch(self, x, y, use_gpu):
"""Initialize a param to init, and compute param -= y."""
with self.cached_session(use_gpu=use_gpu):
with test_util.device(use_gpu=use_gpu):
p = variables.Variable(x)
sub = state_ops.assign_sub(p, y)
self.evaluate(p.initializer)
@ -78,11 +78,10 @@ class AssignOpTest(test.TestCase):
var_value, op_value = self._initAssignAddFetch(x, y, use_gpu=True)
self.assertAllEqual(x + y, var_value)
self.assertAllEqual(x + y, op_value)
var_value, op_value = self._initAssignSubFetch(x, y, use_gpu=False)
var_value, op_value = self._initAssignSubFetch(x, y, use_gpu=True)
self.assertAllEqual(x - y, var_value)
self.assertAllEqual(x - y, op_value)
@test_util.run_deprecated_v1
def testBasic(self):
self._testTypes(np.arange(0, 20).reshape([4, 5]))