Update tf.concat_v2 to tf.concat

320545f0 · Neal Wu · e2ecda2c · 320545f0
Commit 320545f0 authored Jan 30, 2017 by Neal Wu
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tutorials/image/cifar10/cifar10_multi_gpu_train.py tutorials/image/cifar10/cifar10_multi_gpu_train.py +1 -1

No files found.
--- a/tutorials/image/cifar10/cifar10_multi_gpu_train.py
+++ b/tutorials/image/cifar10/cifar10_multi_gpu_train.py
@@ -124,7 +124,7 @@ def average_gradients(tower_grads):
      grads.append(expanded_g)

    # Average over the 'tower' dimension.
-    grad = tf.concat_v2(grads, 0)
+    grad = tf.concat(grads, 0)
    grad = tf.reduce_mean(grad, 0)

    # Keep in mind that the Variables are redundant because they are shared