diff options
author | Alexandre Passos <apassos@google.com> | 2018-09-27 13:18:33 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-27 13:23:04 -0700 |
commit | 4cedc8b6e738b7a188c9c091cf667bacafae44b7 (patch) | |
tree | 56de35940e5f9daedd5f39a82d2cd90cf374e4e4 /tensorflow/contrib/training | |
parent | c898e63d07fc63315be98f0772736e5d7f2fb44c (diff) |
Updating the V2 variables API.
PiperOrigin-RevId: 214824023
Diffstat (limited to 'tensorflow/contrib/training')
-rw-r--r-- | tensorflow/contrib/training/python/training/device_setter_test.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tensorflow/contrib/training/python/training/device_setter_test.py b/tensorflow/contrib/training/python/training/device_setter_test.py index 20746d911c..3bb2dce83d 100644 --- a/tensorflow/contrib/training/python/training/device_setter_test.py +++ b/tensorflow/contrib/training/python/training/device_setter_test.py @@ -98,10 +98,10 @@ class GreedyLoadBalancingStrategyTest(test.TestCase): cluster=_CLUSTER_SPEC, ps_strategy=device_setter_lib.GreedyLoadBalancingStrategy( 2, device_setter_lib.byte_size_load_fn))): - u = variables.Variable(array_ops.zeros([2, 2])) - v = variables.Variable(array_ops.zeros([2, 1])) - w = variables.Variable(array_ops.zeros([2, 2])) - x = variables.Variable(array_ops.zeros([1, 3])) + u = variables.VariableV1(array_ops.zeros([2, 2])) + v = variables.VariableV1(array_ops.zeros([2, 1])) + w = variables.VariableV1(array_ops.zeros([2, 2])) + x = variables.VariableV1(array_ops.zeros([1, 3])) a = v + w self.assertDeviceEqual("/job:ps/task:0", u.device) self.assertDeviceEqual("/job:ps/task:0", u.initializer.device) |