diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2016-10-20 13:46:58 -0800 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-10-20 15:07:18 -0700 |
| commit | 5cfb0fd9c7a89a788188f83886a7321b8a819369 (patch) | |
| tree | 97d80a9027b39e97f031391cdb7d6003d89fae28 /tensorflow/core/framework/common_shape_fns.h | |
| parent | 45e11c3658d07bf309c7190d3bd5fa8bc7fce3f7 (diff) | |
Add ConcatV2 operation which is the same as Concat but with argument order
swapped.
Change: 136763054
Diffstat (limited to 'tensorflow/core/framework/common_shape_fns.h')
| -rw-r--r-- | tensorflow/core/framework/common_shape_fns.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/core/framework/common_shape_fns.h b/tensorflow/core/framework/common_shape_fns.h index ad4425a48f..2e711f5277 100644 --- a/tensorflow/core/framework/common_shape_fns.h +++ b/tensorflow/core/framework/common_shape_fns.h @@ -185,6 +185,9 @@ Status ReductionShapeForReduceJoin(shape_inference::InferenceContext* c); // Shape function for concat operations. Status ConcatShape(shape_inference::InferenceContext* c); +// Shape function for concat operations. +Status ConcatV2Shape(shape_inference::InferenceContext* c); + // Shape function for binary operators that broadcast their inputs. // Tested by ops/math_ops_test.cc. Status BroadcastBinaryOpShapeFn(InferenceContext* c); |