diff options
Diffstat (limited to 'tensorflow/python/debug/lib/dist_session_debug_grpc_test.py')
| -rw-r--r-- | tensorflow/python/debug/lib/dist_session_debug_grpc_test.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/python/debug/lib/dist_session_debug_grpc_test.py b/tensorflow/python/debug/lib/dist_session_debug_grpc_test.py index 46a7be5808..74498c8ea3 100644 --- a/tensorflow/python/debug/lib/dist_session_debug_grpc_test.py +++ b/tensorflow/python/debug/lib/dist_session_debug_grpc_test.py @@ -118,8 +118,8 @@ class DistributedSessionDebugTest(test_util.TensorFlowTestCase): """ with ops.Graph().as_default() as graph: with ops.device("/job:worker/task:0/cpu:0"): - self.a = variables.Variable(10.0, name="a") - self.b = variables.Variable(100.0, name="b") + self.a = variables.VariableV1(10.0, name="a") + self.b = variables.VariableV1(100.0, name="b") self.inc_a = state_ops.assign_add(self.a, 2.0, name="inc_a") self.dec_b = state_ops.assign_add(self.b, -5.0, name="dec_b") self.p = math_ops.multiply(self.inc_a, self.dec_b, name="p") |