diff options
author | 2018-03-27 17:02:59 -0700 | |
---|---|---|
committer | 2018-03-27 17:05:17 -0700 | |
commit | 6f8b85d301140ce42c0aa4871750ee0aec758105 (patch) | |
tree | 6a7484d391c060cbe90158b31c979e8d3654f2cd /tensorflow/compiler/xla/service/shape_inference.cc | |
parent | b2644288ebfb8a4cc52231c3ca93a968397c860a (diff) |
[XLA] Redesign: implement Tuple and GetTupleElement.
PiperOrigin-RevId: 190698245
Diffstat (limited to 'tensorflow/compiler/xla/service/shape_inference.cc')
-rw-r--r-- | tensorflow/compiler/xla/service/shape_inference.cc | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/shape_inference.cc b/tensorflow/compiler/xla/service/shape_inference.cc index 36456d552d..77e12d3602 100644 --- a/tensorflow/compiler/xla/service/shape_inference.cc +++ b/tensorflow/compiler/xla/service/shape_inference.cc @@ -1070,6 +1070,12 @@ ShapeInference::InferDegenerateDimensionBroadcastShape( for (const HloInstruction* operand : operands) { operand_shapes.push_back(&operand->shape()); } + return InferVariadicOpShape(opcode, operand_shapes); +} + +/* static */ StatusOr<Shape> ShapeInference::InferVariadicOpShape( + HloOpcode opcode, + tensorflow::gtl::ArraySlice<const Shape*> operand_shapes) { return InferVariadicOpShape(OpcodeToVariadicOperation(opcode), operand_shapes); } |