diff options
Diffstat (limited to 'tensorflow/python/training/training_util_test.py')
-rw-r--r-- | tensorflow/python/training/training_util_test.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/python/training/training_util_test.py b/tensorflow/python/training/training_util_test.py index 6cc177e0e8..ba64e785ac 100644 --- a/tensorflow/python/training/training_util_test.py +++ b/tensorflow/python/training/training_util_test.py @@ -49,7 +49,7 @@ class GlobalStepTest(test.TestCase): def test_invalid_shape(self): with ops.Graph().as_default() as g: self.assertIsNone(training_util.get_global_step()) - variables.Variable( + variables.VariableV1( [0], trainable=False, dtype=dtypes.int32, @@ -73,7 +73,7 @@ class GlobalStepTest(test.TestCase): def test_get_global_step(self): with ops.Graph().as_default() as g: self.assertIsNone(training_util.get_global_step()) - variables.Variable( + variables.VariableV1( 0, trainable=False, dtype=dtypes.int32, |