Merge pull request #966 from tensorflow/update-cifar10

Update tf.concat_v2 to tf.concat

Merge pull request #966 from tensorflow/update-cifar10
Update tf.concat_v2 to tf.concat
2fd3dcf3 · annarev · GitHub · d43335c5 · 320545f0 · 2fd3dcf3
Commit 2fd3dcf3 authored Jan 30, 2017 by annarev Committed by GitHub Jan 30, 2017
Show 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