From 6b4b9d5c16cddebf6a91b0d027848c3c12371982 Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Wed, 14 Mar 2018 11:49:04 -0700
Subject: [XLA] Add XlaBuilder. This is the first step of implementing the
 client-service interface redesign. Implemented ops: Add, Call, Constant,
 Parameter.

PiperOrigin-RevId: 189061445
---
 tensorflow/compiler/xla/service/shape_inference.cc | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'tensorflow/compiler/xla/service/shape_inference.cc')

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
@@ -944,6 +944,13 @@ ShapeInference::InferDegenerateDimensionBroadcastShape(
                             rhs->shape(), /*broadcast_dimensions=*/{});
 }
 
+/* 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) {
-- 
cgit v1.2.3