diff options
author | 2018-03-14 11:49:04 -0700 | |
---|---|---|
committer | 2018-03-14 11:57:18 -0700 | |
commit | 6b4b9d5c16cddebf6a91b0d027848c3c12371982 (patch) | |
tree | 9988a6b13f884ec9aa83804860b2e0b9c985ed78 /tensorflow/compiler/xla/service/shape_inference.h | |
parent | 655358afc90f426c2d7a9daaf81007d203408cdd (diff) |
[XLA] Add XlaBuilder. This is the first step of implementing the client-service interface redesign. Implemented ops: Add, Call, Constant, Parameter.
PiperOrigin-RevId: 189061445
Diffstat (limited to 'tensorflow/compiler/xla/service/shape_inference.h')
-rw-r--r-- | tensorflow/compiler/xla/service/shape_inference.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/shape_inference.h b/tensorflow/compiler/xla/service/shape_inference.h index 0d3045213d..085fdac60c 100644 --- a/tensorflow/compiler/xla/service/shape_inference.h +++ b/tensorflow/compiler/xla/service/shape_inference.h @@ -56,6 +56,9 @@ class ShapeInference { static StatusOr<Shape> InferBinaryOpShape( BinaryOperation operation, const Shape& lhs, const Shape& rhs, tensorflow::gtl::ArraySlice<int64> broadcast_dimensions); + static StatusOr<Shape> InferBinaryOpShape( + HloOpcode opcode, const Shape& lhs, const Shape& rhs, + tensorflow::gtl::ArraySlice<int64> broadcast_dimensions); static StatusOr<Shape> InferBinaryOpShape(HloOpcode opcode, const HloInstruction* lhs, const HloInstruction* rhs); |