diff options
Diffstat (limited to 'tensorflow/contrib/layers/python/layers/feature_column_ops_test.py')
-rw-r--r-- | tensorflow/contrib/layers/python/layers/feature_column_ops_test.py | 8 |
1 files changed, 4 insertions, 4 deletions
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 032ea57cf5..7f95e95c1b 100644 --- a/tensorflow/contrib/layers/python/layers/feature_column_ops_test.py +++ b/tensorflow/contrib/layers/python/layers/feature_column_ops_test.py @@ -1209,7 +1209,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_variables_initializer().run() self.assertAllEqual(logits.eval().shape, [2, 5]) def testWeightedSparseColumn(self): @@ -1244,7 +1244,7 @@ class WeightedSumTest(tf.test.TestCase): features, [weighted_ids], num_outputs=5) with self.test_session(): - tf.initialize_all_variables().run() + tf.global_variables_initializer().run() tf.initialize_all_tables().run() self.assertAllEqual(logits.eval().shape, [2, 5]) @@ -1846,7 +1846,7 @@ class WeightedSumTest(tf.test.TestCase): [product], num_outputs=1)) with self.test_session() as sess: - tf.initialize_all_variables().run() + tf.global_variables_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]])) @@ -1862,7 +1862,7 @@ class WeightedSumTest(tf.test.TestCase): [product], num_outputs=1)) with self.test_session() as sess: - tf.initialize_all_variables().run() + tf.global_variables_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]])) |