Searched refs:InferElementwiseBinaryOpShape (Results 1 – 2 of 2) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | shape_inference.cc | 899 /* static */ StatusOr<Shape> ShapeInference::InferElementwiseBinaryOpShape( in InferElementwiseBinaryOpShape() function in xla::ShapeInference 982 return InferElementwiseBinaryOpShape(opcode, lhs, rhs, in InferBinaryOpShape() 999 return InferElementwiseBinaryOpShape(opcode, lhs, rhs, in InferBinaryOpShape() 1010 InferElementwiseBinaryOpShape(opcode, lhs, rhs, in InferBinaryOpShape() 1030 return InferElementwiseBinaryOpShape(opcode, lhs, rhs, in InferBinaryOpShape() 1034 InferElementwiseBinaryOpShape(opcode, lhs, rhs, in InferBinaryOpShape()
|
H A D | shape_inference.h | 382 static StatusOr<Shape> InferElementwiseBinaryOpShape(
|