diff options
author | 2016-12-13 22:41:36 -0800 | |
---|---|---|
committer | 2016-12-13 23:05:56 -0800 | |
commit | 2cf68e3e070d1a96bc7caa38f34e9c7bcb25e612 (patch) | |
tree | cec848c74f256c86a29fb3476c6fc68b88e35498 | |
parent | 96e24a11400a3604a47b7918c1f9ae5909794a4d (diff) |
Switch leftover reference to tf.concat to tf.concat_v2 in array_grad.py.
Change: 141981559
-rw-r--r-- | tensorflow/python/ops/array_grad.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/python/ops/array_grad.py b/tensorflow/python/ops/array_grad.py index e657478474..f828490647 100644 --- a/tensorflow/python/ops/array_grad.py +++ b/tensorflow/python/ops/array_grad.py @@ -307,7 +307,7 @@ def _MatrixSetDiagGrad(op, grad): batch_shape = array_ops.slice(grad_shape, [0], [grad_rank - 2]) matrix_shape = array_ops.slice(grad_shape, [grad_rank - 2], [2]) min_dim = math_ops.reduce_min(matrix_shape) - diag_shape = array_ops.concat(0, [batch_shape, [min_dim]]) + diag_shape = array_ops.concat_v2([batch_shape, [min_dim]], 0) grad_input = array_ops.matrix_set_diag( grad, array_ops.zeros( diag_shape, dtype=grad.dtype)) @@ -345,7 +345,7 @@ def _GatherGrad(op, grad): # Build appropriately shaped IndexedSlices indices = op.inputs[1] size = array_ops.expand_dims(array_ops.size(indices), 0) - values_shape = array_ops.concat(0, [size, params_shape[1:]]) + values_shape = array_ops.concat_v2([size, params_shape[1:]], 0) values = array_ops.reshape(grad, values_shape) indices = array_ops.reshape(indices, size) return [ops.IndexedSlices(values, indices, params_shape), None] |