Home
last modified time | relevance | path

Searched refs:InferBinaryOpShape (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dshape_inference_test.cc270 return ShapeInference::InferBinaryOpShape(HloOpcode::kComplex, lhs, rhs, in TEST_F()
1397 auto inferred_status = ShapeInference::InferBinaryOpShape( in TEST_F()
1405 auto inferred_status = ShapeInference::InferBinaryOpShape( in TEST_F()
1864 ShapeInference::InferBinaryOpShape(HloOpcode::kAdd, mat, vec8, {1}); in TEST_F()
1869 ShapeInference::InferBinaryOpShape(HloOpcode::kAdd, mat, vec8, {0}); in TEST_F()
1873 ShapeInference::InferBinaryOpShape(HloOpcode::kAdd, mat, vec16, {0}); in TEST_F()
1878 ShapeInference::InferBinaryOpShape(HloOpcode::kAdd, mat, vec16, {1}); in TEST_F()
1889 auto inferred_status_match = ShapeInference::InferBinaryOpShape( in TEST_F()
1894 inferred_status_match = ShapeInference::InferBinaryOpShape( in TEST_F()
1899 inferred_status_match = ShapeInference::InferBinaryOpShape( in TEST_F()
[all …]
H A Dshape_inference.h54 static StatusOr<Shape> InferBinaryOpShape(
57 static StatusOr<Shape> InferBinaryOpShape(HloOpcode opcode,
H A Ddynamic_window_utils.cc54 ShapeInference::InferBinaryOpShape(opcode, x.get(), y.get()).ValueOrDie(); in BinaryOp()
H A Dhlo_creation_utils.cc66 ShapeInference::InferBinaryOpShape(opcode, lhs, rhs)); in MakeBinaryHlo()
80 ShapeInference::InferBinaryOpShape(HloOpcode::kCompare, lhs, rhs)); in MakeCompareHlo()
H A Dshape_inference.cc955 /* static */ StatusOr<Shape> ShapeInference::InferBinaryOpShape( in InferBinaryOpShape() function in xla::ShapeInference
957 return InferBinaryOpShape(opcode, lhs->shape(), rhs->shape(), in InferBinaryOpShape()
961 /* static */ StatusOr<Shape> ShapeInference::InferBinaryOpShape( in InferBinaryOpShape() function in xla::ShapeInference
H A Dhlo_verifier.cc1791 instruction, ShapeInference::InferBinaryOpShape(instruction->opcode(), in CheckBinaryShape()
H A Dhlo_parser.cc1382 return ShapeInference::InferBinaryOpShape(opcode, operands[0], in CreateInstruction()
2081 return ShapeInference::InferBinaryOpShape(opcode, operands[0], in CreateInstruction()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dxla_builder.cc696 Shape shape, ShapeInference::InferBinaryOpShape( in BinaryOp()