Searched refs:InferBinaryOpShape (Results 1 – 8 of 8) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | shape_inference_test.cc | 270 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 D | shape_inference.h | 54 static StatusOr<Shape> InferBinaryOpShape( 57 static StatusOr<Shape> InferBinaryOpShape(HloOpcode opcode,
|
H A D | dynamic_window_utils.cc | 54 ShapeInference::InferBinaryOpShape(opcode, x.get(), y.get()).ValueOrDie(); in BinaryOp()
|
H A D | hlo_creation_utils.cc | 66 ShapeInference::InferBinaryOpShape(opcode, lhs, rhs)); in MakeBinaryHlo() 80 ShapeInference::InferBinaryOpShape(HloOpcode::kCompare, lhs, rhs)); in MakeCompareHlo()
|
H A D | shape_inference.cc | 955 /* 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 D | hlo_verifier.cc | 1791 instruction, ShapeInference::InferBinaryOpShape(instruction->opcode(), in CheckBinaryShape()
|
H A D | hlo_parser.cc | 1382 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 D | xla_builder.cc | 696 Shape shape, ShapeInference::InferBinaryOpShape( in BinaryOp()
|