diff options
author | 2018-05-29 21:24:36 -0700 | |
---|---|---|
committer | 2018-05-29 21:27:20 -0700 | |
commit | 9c509eedc3888d3846b2ab5ac2879268df9ff8cd (patch) | |
tree | 07a597f1409eaea8c38d7039e6580ff0f09e1b09 /tensorflow/compiler/xla/service/shape_inference.cc | |
parent | 3f2ba2edf62dc394cfcb4b2606f1638389aa92e2 (diff) |
Introduced kDomain HLO instruction set isolation to bound connected sets of instructions with similar attributes (ie, sharding).
This CL simply adds the infrastructure, but leaves the wire-on to a separate CL.
PiperOrigin-RevId: 198503625
Diffstat (limited to 'tensorflow/compiler/xla/service/shape_inference.cc')
-rw-r--r-- | tensorflow/compiler/xla/service/shape_inference.cc | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/compiler/xla/service/shape_inference.cc b/tensorflow/compiler/xla/service/shape_inference.cc index 3500978bdd..d624f548b1 100644 --- a/tensorflow/compiler/xla/service/shape_inference.cc +++ b/tensorflow/compiler/xla/service/shape_inference.cc @@ -316,7 +316,8 @@ StatusOr<Shape> InferWindowOutputShape(const Shape& base_shape, /* static */ StatusOr<Shape> ShapeInference::InferUnaryOpShape( HloOpcode opcode, const Shape& shape) { // There is no copy operation at the proto level, so handle copy explicitly. - if (opcode == HloOpcode::kCopy) { + // A domain shape is the same as the input one. + if (opcode == HloOpcode::kCopy || opcode == HloOpcode::kDomain) { return shape; } |