diff options
author | 2016-12-06 18:25:37 -0800 | |
---|---|---|
committer | 2016-12-06 18:44:18 -0800 | |
commit | d4eb834824d79c6a64a3c4a1c4a88b434b73e63e (patch) | |
tree | 3a6a417a668e79bc588929450f1f7794bb9eee2c /tensorflow/contrib/labeled_tensor | |
parent | 7b306e8fcfb6db3f438c27e437194e78c1d73e23 (diff) |
Switch all tf.concat(concat_dim, value, name) calls in third_party/tensorflow to tf.concat_v2(value, axis, name).
Change: 141255675
Diffstat (limited to 'tensorflow/contrib/labeled_tensor')
-rw-r--r-- | tensorflow/contrib/labeled_tensor/python/ops/ops.py | 2 | ||||
-rw-r--r-- | tensorflow/contrib/labeled_tensor/python/ops/ops_test.py | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/labeled_tensor/python/ops/ops.py b/tensorflow/contrib/labeled_tensor/python/ops/ops.py index d846b013fe..f4b83ed5e6 100644 --- a/tensorflow/contrib/labeled_tensor/python/ops/ops.py +++ b/tensorflow/contrib/labeled_tensor/python/ops/ops.py @@ -198,7 +198,7 @@ def concat(labeled_tensors, axis_name, name=None): concat_axis = core.concat_axes(concat_axis_list) concat_dimension = axis_names.index(axis_name) - concat_tensor = array_ops.concat(concat_dimension, tensors, name=scope) + concat_tensor = array_ops.concat_v2(tensors, concat_dimension, name=scope) values = list(axes_0.values()) concat_axes = (values[:concat_dimension] + [concat_axis] + values[concat_dimension + 1:]) diff --git a/tensorflow/contrib/labeled_tensor/python/ops/ops_test.py b/tensorflow/contrib/labeled_tensor/python/ops/ops_test.py index c19fc09f93..55dd96e560 100644 --- a/tensorflow/contrib/labeled_tensor/python/ops/ops_test.py +++ b/tensorflow/contrib/labeled_tensor/python/ops/ops_test.py @@ -903,7 +903,7 @@ class WhereTest(Base): where_lt = ops.where(condition, x, y) golden_lt = core.LabeledTensor( - tf.concat(0, [tf.ones(3), tf.zeros(2)]), ['x']) + tf.concat_v2([tf.ones(3), tf.zeros(2)], 0), ['x']) self.assertLabeledTensorsEqual(where_lt, golden_lt) def test_mismatched_axes(self): |