aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/training/supervisor_test.py
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/python/training/supervisor_test.py')
-rw-r--r--tensorflow/python/training/supervisor_test.py8
1 files changed, 4 insertions, 4 deletions
diff --git a/tensorflow/python/training/supervisor_test.py b/tensorflow/python/training/supervisor_test.py
index dda0166aa6..c7c16cdf81 100644
--- a/tensorflow/python/training/supervisor_test.py
+++ b/tensorflow/python/training/supervisor_test.py
@@ -531,7 +531,7 @@ class SupervisorTest(tf.test.TestCase):
collections=[tf.GraphKeys.LOCAL_VARIABLES],
name="default_ready_for_local_init_op_w_" + str(uid))
ready_for_local_init_op = tf.report_uninitialized_variables(
- tf.all_variables())
+ tf.global_variables())
sv = tf.train.Supervisor(
logdir=logdir,
is_chief=is_chief,
@@ -588,7 +588,7 @@ class SupervisorTest(tf.test.TestCase):
collections=[tf.GraphKeys.LOCAL_VARIABLES],
name="ready_for_local_init_op_restore_w_" + str(uid))
ready_for_local_init_op = tf.report_uninitialized_variables(
- tf.all_variables())
+ tf.global_variables())
sv = tf.train.Supervisor(
logdir=logdir,
is_chief=is_chief,
@@ -624,7 +624,7 @@ class SupervisorTest(tf.test.TestCase):
# This shouldn't add a variable to the VARIABLES collection responsible
# for variables that are saved/restored from checkpoints.
- self.assertEquals(len(tf.all_variables()), 0)
+ self.assertEquals(len(tf.global_variables()), 0)
# Suppress normal variable inits to make sure the local one is
# initialized via local_init_op.
@@ -644,7 +644,7 @@ class SupervisorTest(tf.test.TestCase):
collections=[tf.GraphKeys.LOCAL_VARIABLES])
# This shouldn't add a variable to the VARIABLES collection responsible
# for variables that are saved/restored from checkpoints.
- self.assertEquals(len(tf.all_variables()), 0)
+ self.assertEquals(len(tf.global_variables()), 0)
# Suppress normal variable inits to make sure the local one is
# initialized via local_init_op.