diff options
author | 2018-09-27 13:18:33 -0700 | |
---|---|---|
committer | 2018-09-27 13:23:04 -0700 | |
commit | 4cedc8b6e738b7a188c9c091cf667bacafae44b7 (patch) | |
tree | 56de35940e5f9daedd5f39a82d2cd90cf374e4e4 /tensorflow/python/training/learning_rate_decay_test.py | |
parent | c898e63d07fc63315be98f0772736e5d7f2fb44c (diff) |
Updating the V2 variables API.
PiperOrigin-RevId: 214824023
Diffstat (limited to 'tensorflow/python/training/learning_rate_decay_test.py')
-rw-r--r-- | tensorflow/python/training/learning_rate_decay_test.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/python/training/learning_rate_decay_test.py b/tensorflow/python/training/learning_rate_decay_test.py index 5a9215730e..03a32f6ca0 100644 --- a/tensorflow/python/training/learning_rate_decay_test.py +++ b/tensorflow/python/training/learning_rate_decay_test.py @@ -63,7 +63,7 @@ class LRDecayTest(test_util.TensorFlowTestCase): def testVariables(self): with self.cached_session(): - step = variables.Variable(1) + step = variables.VariableV1(1) assign_1 = step.assign(1) assign_2 = step.assign(2) assign_100 = step.assign(100) @@ -121,7 +121,7 @@ class LRDecayTest(test_util.TensorFlowTestCase): # Test that ref types are valid. if not context.executing_eagerly(): - x = variables.Variable(0.0) + x = variables.VariableV1(0.0) x_ref = x.op.outputs[0] # float32_ref tensor should be accepted boundaries, values = [1.0, 2.0], [1, 2, 3] learning_rate_decay.piecewise_constant(x_ref, boundaries, values) |