diff options
author | 2016-12-04 22:08:45 -0800 | |
---|---|---|
committer | 2016-12-04 22:08:45 -0800 | |
commit | 83d33cc0be732ad526785922391a6ec92f9a8801 (patch) | |
tree | 639ea317242525952f60d289eb63e643a4f93b06 | |
parent | 32b584d2bad23b9393111ca8ec28888f888bdf18 (diff) |
Replace deprecated functions in contrib, models
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): |