diff options
Diffstat (limited to 'tensorflow/contrib/layers/python/layers/optimizers_test.py')
-rw-r--r-- | tensorflow/contrib/layers/python/layers/optimizers_test.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/layers/python/layers/optimizers_test.py b/tensorflow/contrib/layers/python/layers/optimizers_test.py index 1dfed82103..ab183ba75d 100644 --- a/tensorflow/contrib/layers/python/layers/optimizers_test.py +++ b/tensorflow/contrib/layers/python/layers/optimizers_test.py @@ -195,7 +195,7 @@ class OptimizersTest(tf.test.TestCase): self.assertAlmostEqual(var_value, 9.8916, 4) self.assertEqual(global_step_value, 1) var_count = 0 - for var in tf.all_variables(): + for var in tf.global_variables(): if var.name.startswith("OptimizeLoss/AdaptiveMaxNorm"): var_count += 1 self.assertEqual(2, var_count) @@ -366,7 +366,7 @@ class AdaptiveClipping(tf.test.TestCase): decay=0.5)(grads_and_vars) var_dict = {} - for var in tf.all_variables(): + for var in tf.global_variables(): if var.name.startswith("AdaptiveMaxNorm"): var_dict[var.name.split(":")[0]] = var self.assertEqual(2, len(var_dict)) |