diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-03-14 11:49:04 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-03-14 11:57:18 -0700 |
commit | 6b4b9d5c16cddebf6a91b0d027848c3c12371982 (patch) | |
tree | 9988a6b13f884ec9aa83804860b2e0b9c985ed78 /tensorflow/compiler/xla/service/shape_inference.cc | |
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.cc')
-rw-r--r-- | tensorflow/compiler/xla/service/shape_inference.cc | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/shape_inference.cc b/tensorflow/compiler/xla/service/shape_inference.cc index 74f744a62b..8c8bd6d73a 100644 --- a/tensorflow/compiler/xla/service/shape_inference.cc +++ b/tensorflow/compiler/xla/service/shape_inference.cc @@ -945,6 +945,13 @@ ShapeInference::InferDegenerateDimensionBroadcastShape( } /* static */ StatusOr<Shape> ShapeInference::InferBinaryOpShape( + HloOpcode opcode, const Shape& lhs, const Shape& rhs, + tensorflow::gtl::ArraySlice<int64> broadcast_dimensions) { + return InferBinaryOpShape(OpcodeToBinaryOperation(opcode), lhs, rhs, + broadcast_dimensions); +} + +/* static */ StatusOr<Shape> ShapeInference::InferBinaryOpShape( BinaryOperation operation, const Shape& lhs, const Shape& rhs, tensorflow::gtl::ArraySlice<int64> broadcast_dimensions) { VLOG(2) << tensorflow::strings::Printf( |