aboutsummaryrefslogtreecommitdiffhomepage
diff options
context:
space:
mode:
authorGravatar Sam Abrahams <sjabraha@gmail.com>2016-12-04 22:08:45 -0800
committerGravatar Sam Abrahams <sjabraha@gmail.com>2016-12-04 22:08:45 -0800
commit83d33cc0be732ad526785922391a6ec92f9a8801 (patch)
tree639ea317242525952f60d289eb63e643a4f93b06
parent32b584d2bad23b9393111ca8ec28888f888bdf18 (diff)
Replace deprecated functions in contrib, models
-rw-r--r--tensorflow/contrib/bayesflow/python/kernel_tests/stochastic_variables_test.py8
-rw-r--r--tensorflow/contrib/framework/python/ops/prettyprint_ops_test.py2
-rw-r--r--tensorflow/contrib/layers/python/layers/feature_column_ops_test.py8
-rw-r--r--tensorflow/contrib/layers/python/layers/optimizers_test.py4
-rw-r--r--tensorflow/contrib/rnn/python/kernel_tests/gru_ops_test.py6
-rw-r--r--tensorflow/contrib/rnn/python/kernel_tests/rnn_test.py2
-rw-r--r--tensorflow/contrib/seq2seq/python/kernel_tests/seq2seq_test.py2
-rw-r--r--tensorflow/contrib/slim/README.md2
-rw-r--r--tensorflow/contrib/slim/python/slim/learning_test.py4
-rw-r--r--tensorflow/contrib/slim/python/slim/model_analyzer.py2
-rw-r--r--tensorflow/contrib/specs/python/specs_test.py4
-rw-r--r--tensorflow/contrib/stat_summarizer/python/stat_summarizer_test.py2
-rw-r--r--tensorflow/contrib/training/python/training/evaluation_test.py4
-rw-r--r--tensorflow/contrib/training/python/training/training_test.py2
-rw-r--r--tensorflow/models/image/cifar10/cifar10_multi_gpu_train.py2
-rw-r--r--tensorflow/models/rnn/translate/seq2seq_model.py2
16 files changed, 28 insertions, 28 deletions
diff --git a/tensorflow/contrib/bayesflow/python/kernel_tests/stochastic_variables_test.py b/tensorflow/contrib/bayesflow/python/kernel_tests/stochastic_variables_test.py
index 1646abcd9f..74bf699d22 100644
--- a/tensorflow/contrib/bayesflow/python/kernel_tests/stochastic_variables_test.py
+++ b/tensorflow/contrib/bayesflow/python/kernel_tests/stochastic_variables_test.py
@@ -42,8 +42,8 @@ class StochasticVariablesTest(tf.test.TestCase):
self.assertEqual(
{"stochastic_variables/sv_mu", "stochastic_variables/sv_sigma"},
- set([v.op.name for v in tf.all_variables()]))
- self.assertEqual(set(tf.trainable_variables()), set(tf.all_variables()))
+ set([v.op.name for v in tf.global_variables()]))
+ self.assertEqual(set(tf.trainable_variables()), set(tf.global_variables()))
v = tf.convert_to_tensor(v)
self.assertEqual(list(shape), v.get_shape().as_list())
@@ -64,7 +64,7 @@ class StochasticVariablesTest(tf.test.TestCase):
})):
v = tf.get_variable("sv")
- for var in tf.all_variables():
+ for var in tf.global_variables():
if "mu" in var.name:
mu_var = var
if "sigma" in var.name:
@@ -96,7 +96,7 @@ class StochasticVariablesTest(tf.test.TestCase):
})):
v = tf.get_variable("sv", shape)
- for var in tf.all_variables():
+ for var in tf.global_variables():
if "mu" in var.name:
mu_var = var
if "sigma" in var.name:
diff --git a/tensorflow/contrib/framework/python/ops/prettyprint_ops_test.py b/tensorflow/contrib/framework/python/ops/prettyprint_ops_test.py
index bcea8692a6..d7419738fc 100644
--- a/tensorflow/contrib/framework/python/ops/prettyprint_ops_test.py
+++ b/tensorflow/contrib/framework/python/ops/prettyprint_ops_test.py
@@ -50,7 +50,7 @@ class PrettyPrintOpsTest(tf.test.TestCase):
a = tf.Variable(1.0)
a = tf.contrib.framework.print_op(a)
with self.test_session():
- tf.initialize_all_variables().run()
+ tf.global_variable_initializer().run()
a.eval()
if __name__ == "__main__":
diff --git a/tensorflow/contrib/layers/python/layers/feature_column_ops_test.py b/tensorflow/contrib/layers/python/layers/feature_column_ops_test.py
index 8a49e14c08..b5d7ec4154 100644
--- a/tensorflow/contrib/layers/python/layers/feature_column_ops_test.py
+++ b/tensorflow/contrib/layers/python/layers/feature_column_ops_test.py
@@ -1207,7 +1207,7 @@ class WeightedSumTest(tf.test.TestCase):
logits, _, _ = tf.contrib.layers.weighted_sum_from_feature_columns(
features, [hashed_sparse], num_outputs=5)
with self.test_session():
- tf.initialize_all_variables().run()
+ tf.global_variable_initializer().run()
self.assertAllEqual(logits.eval().shape, [2, 5])
def testWeightedSparseColumn(self):
@@ -1242,7 +1242,7 @@ class WeightedSumTest(tf.test.TestCase):
features, [weighted_ids], num_outputs=5)
with self.test_session():
- tf.initialize_all_variables().run()
+ tf.global_variable_initializer().run()
tf.initialize_all_tables().run()
self.assertAllEqual(logits.eval().shape, [2, 5])
@@ -1844,7 +1844,7 @@ class WeightedSumTest(tf.test.TestCase):
[product],
num_outputs=1))
with self.test_session() as sess:
- tf.initialize_all_variables().run()
+ tf.global_variable_initializer().run()
tf.initialize_all_tables().run()
product_weights = column_to_variable[product][0]
sess.run(product_weights.assign([[0.1], [0.2], [0.3], [0.4], [0.5]]))
@@ -1860,7 +1860,7 @@ class WeightedSumTest(tf.test.TestCase):
[product],
num_outputs=1))
with self.test_session() as sess:
- tf.initialize_all_variables().run()
+ tf.global_variable_initializer().run()
tf.initialize_all_tables().run()
product_weights = column_to_variable[product][0]
sess.run(product_weights.assign([[0.1], [0.2], [0.3], [0.4], [0.5]]))
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))
diff --git a/tensorflow/contrib/rnn/python/kernel_tests/gru_ops_test.py b/tensorflow/contrib/rnn/python/kernel_tests/gru_ops_test.py
index 9f008023bf..0ca4bf0df0 100644
--- a/tensorflow/contrib/rnn/python/kernel_tests/gru_ops_test.py
+++ b/tensorflow/contrib/rnn/python/kernel_tests/gru_ops_test.py
@@ -158,7 +158,7 @@ class GRUBlockCellTest(tf.test.TestCase):
output = gru_ops.GRUBlockCell(cell_size)(x, h)
sess.run([tf.global_variables_initializer()])
- all_variables = tf.all_variables()[0:4]
+ all_variables = tf.global_variables()[0:4]
[w_ru, b_ru, w_c, b_c] = all_variables
d_new_h_wrt_x = tf.gradients([output], x)
@@ -178,7 +178,7 @@ class GRUBlockCellTest(tf.test.TestCase):
output = tf.nn.rnn_cell.GRUCell(cell_size)(x, h)
sess.run([tf.global_variables_initializer()])
- all_variables = tf.all_variables()[4:8]
+ all_variables = tf.global_variables()[4:8]
[w_ru, b_ru, w_c, b_c] = all_variables
d_new_h_wrt_x = tf.gradients([output], x)
@@ -281,7 +281,7 @@ class GRUBlockCellTest(tf.test.TestCase):
sess.run([tf.global_variables_initializer()])
- all_variables = tf.all_variables()
+ all_variables = tf.global_variables()
[w_ru, b_ru, w_c, b_c] = all_variables[:4]
diff --git a/tensorflow/contrib/rnn/python/kernel_tests/rnn_test.py b/tensorflow/contrib/rnn/python/kernel_tests/rnn_test.py
index 91b3d7f417..486f88e1f7 100644
--- a/tensorflow/contrib/rnn/python/kernel_tests/rnn_test.py
+++ b/tensorflow/contrib/rnn/python/kernel_tests/rnn_test.py
@@ -382,7 +382,7 @@ class StackBidirectionalRNNTest(tf.test.TestCase):
# check that all the variables names starts with the proper scope.
tf.global_variables_initializer()
- all_vars = tf.all_variables()
+ all_vars = tf.global_variables()
prefix = prefix or "stack_bidirectional_rnn"
scope_vars = [v for v in all_vars if v.name.startswith(prefix + "/")]
tf.logging.info("StackRNN with scope: %s (%s)"
diff --git a/tensorflow/contrib/seq2seq/python/kernel_tests/seq2seq_test.py b/tensorflow/contrib/seq2seq/python/kernel_tests/seq2seq_test.py
index a5bb331f81..e9241c425b 100644
--- a/tensorflow/contrib/seq2seq/python/kernel_tests/seq2seq_test.py
+++ b/tensorflow/contrib/seq2seq/python/kernel_tests/seq2seq_test.py
@@ -103,7 +103,7 @@ class Seq2SeqTest(tf.test.TestCase):
scope=scope))
# Run model
- tf.initialize_all_variables().run()
+ tf.global_variable_initializer().run()
decoder_outputs_train_res, decoder_state_train_res = sess.run(
[decoder_outputs_train, decoder_state_train])
decoder_outputs_inference_res, decoder_state_inference_res = sess.run(
diff --git a/tensorflow/contrib/slim/README.md b/tensorflow/contrib/slim/README.md
index 8454ddc2ec..e6100ef675 100644
--- a/tensorflow/contrib/slim/README.md
+++ b/tensorflow/contrib/slim/README.md
@@ -901,7 +901,7 @@ slim.evaluation.evaluation_loop(
log_dir,
num_evals=num_batches,
eval_op=names_to_updates.values(),
- summary_op=tf.merge_summary(summary_ops),
+ summary_op=tf.summary.merge(summary_ops),
eval_interval_secs=eval_interval_secs)
```
diff --git a/tensorflow/contrib/slim/python/slim/learning_test.py b/tensorflow/contrib/slim/python/slim/learning_test.py
index 42949e2c28..7225ab86c4 100644
--- a/tensorflow/contrib/slim/python/slim/learning_test.py
+++ b/tensorflow/contrib/slim/python/slim/learning_test.py
@@ -625,7 +625,7 @@ class TrainTest(tf.test.TestCase):
tf.set_random_seed(2)
train_op = self.create_train_op()
- model_variables = tf.all_variables()
+ model_variables = tf.global_variables()
model_path = os.path.join(logdir1, 'model.ckpt-300')
init_op = tf.global_variables_initializer()
@@ -674,7 +674,7 @@ class TrainTest(tf.test.TestCase):
tf.set_random_seed(2)
train_op = self.create_train_op()
- model_variables = tf.all_variables()
+ model_variables = tf.global_variables()
model_path = os.path.join(logdir1, 'model.ckpt-300')
saver = tf.train.Saver(model_variables)
def RestoreFn(sess):
diff --git a/tensorflow/contrib/slim/python/slim/model_analyzer.py b/tensorflow/contrib/slim/python/slim/model_analyzer.py
index e29c7b1d8c..74617928a7 100644
--- a/tensorflow/contrib/slim/python/slim/model_analyzer.py
+++ b/tensorflow/contrib/slim/python/slim/model_analyzer.py
@@ -84,7 +84,7 @@ def analyze_vars(variables, print_info=False):
"""Prints the names and shapes of the variables.
Args:
- variables: list of variables, for example tf.all_variables().
+ variables: list of variables, for example tf.global_variables().
print_info: Optional, if true print variables and their shape.
Returns:
diff --git a/tensorflow/contrib/specs/python/specs_test.py b/tensorflow/contrib/specs/python/specs_test.py
index a25532ab41..aab64d6b5a 100644
--- a/tensorflow/contrib/specs/python/specs_test.py
+++ b/tensorflow/contrib/specs/python/specs_test.py
@@ -197,7 +197,7 @@ class SpecsTest(tf.test.TestCase):
initializer=tf.constant_initializer(42.0))
inputs = tf.constant(_rand(10, 100))
outputs = v.funcall(inputs)
- self.assertEqual(len(tf.all_variables()), 1)
+ self.assertEqual(len(tf.global_variables()), 1)
sess.run([outputs.initializer])
outputs_value = outputs.eval()
self.assertEqual(outputs_value.shape, (2, 2))
@@ -211,7 +211,7 @@ class SpecsTest(tf.test.TestCase):
g = f | f | f | f
inputs = tf.constant(_rand(10, 100))
_ = g.funcall(inputs)
- self.assertEqual(len(tf.all_variables()), 2)
+ self.assertEqual(len(tf.global_variables()), 2)
def testAutoFunction(self):
with self.test_session():
diff --git a/tensorflow/contrib/stat_summarizer/python/stat_summarizer_test.py b/tensorflow/contrib/stat_summarizer/python/stat_summarizer_test.py
index 616be81e27..312cb5c577 100644
--- a/tensorflow/contrib/stat_summarizer/python/stat_summarizer_test.py
+++ b/tensorflow/contrib/stat_summarizer/python/stat_summarizer_test.py
@@ -34,7 +34,7 @@ class StatSummarizerTest(tf.test.TestCase):
graph_def.SerializeToString())
with self.test_session() as sess:
- sess.run(tf.initialize_all_variables())
+ sess.run(tf.global_variable_initializer())
for _ in range(20):
run_metadata = tf.RunMetadata()
diff --git a/tensorflow/contrib/training/python/training/evaluation_test.py b/tensorflow/contrib/training/python/training/evaluation_test.py
index 927f6ab75a..39889fb043 100644
--- a/tensorflow/contrib/training/python/training/evaluation_test.py
+++ b/tensorflow/contrib/training/python/training/evaluation_test.py
@@ -51,7 +51,7 @@ class CheckpointIteratorTest(tf.test.TestCase):
saver = tf.train.Saver() # Saves the global step.
with self.test_session() as session:
- session.run(tf.initialize_all_variables())
+ session.run(tf.global_variable_initializer())
save_path = os.path.join(checkpoint_dir, 'model.ckpt')
saver.save(session, save_path, global_step=global_step)
@@ -81,7 +81,7 @@ class CheckpointIteratorTest(tf.test.TestCase):
target='',
config=tf.ConfigProto(device_count={'CPU': 2})) as session:
- session.run(tf.initialize_all_variables())
+ session.run(tf.global_variable_initializer())
save_path = os.path.join(checkpoint_dir, 'model.ckpt')
saver.save(session, save_path, global_step=global_step)
diff --git a/tensorflow/contrib/training/python/training/training_test.py b/tensorflow/contrib/training/python/training/training_test.py
index 918c1da018..c0e79aa798 100644
--- a/tensorflow/contrib/training/python/training/training_test.py
+++ b/tensorflow/contrib/training/python/training/training_test.py
@@ -310,7 +310,7 @@ class TrainTest(tf.test.TestCase):
tf.set_random_seed(2)
train_op = self.create_train_op()
- model_variables = tf.all_variables()
+ model_variables = tf.global_variables()
model_path = os.path.join(logdir1, 'model.ckpt-300')
assign_fn = tf.contrib.framework.assign_from_checkpoint_fn(
diff --git a/tensorflow/models/image/cifar10/cifar10_multi_gpu_train.py b/tensorflow/models/image/cifar10/cifar10_multi_gpu_train.py
index a59e13d5e3..c2d1e73f87 100644
--- a/tensorflow/models/image/cifar10/cifar10_multi_gpu_train.py
+++ b/tensorflow/models/image/cifar10/cifar10_multi_gpu_train.py
@@ -213,7 +213,7 @@ def train():
train_op = tf.group(apply_gradient_op, variables_averages_op)
# Create a saver.
- saver = tf.train.Saver(tf.all_variables())
+ saver = tf.train.Saver(tf.global_variables())
# Build the summary operation from the last tower summaries.
summary_op = tf.contrib.deprecated.merge_summary(summaries)
diff --git a/tensorflow/models/rnn/translate/seq2seq_model.py b/tensorflow/models/rnn/translate/seq2seq_model.py
index 23e256c57b..5db351bf25 100644
--- a/tensorflow/models/rnn/translate/seq2seq_model.py
+++ b/tensorflow/models/rnn/translate/seq2seq_model.py
@@ -185,7 +185,7 @@ class Seq2SeqModel(object):
self.updates.append(opt.apply_gradients(
zip(clipped_gradients, params), global_step=self.global_step))
- self.saver = tf.train.Saver(tf.all_variables())
+ self.saver = tf.train.Saver(tf.global_variables())
def step(self, session, encoder_inputs, decoder_inputs, target_weights,
bucket_id, forward_only):