Searched refs:InferBinaryOpShape (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference_test.cc | 256 return ShapeInference::InferBinaryOpShape(HloOpcode::kComplex, lhs, rhs, in TEST_F()
915 auto inferred_status = ShapeInference::InferBinaryOpShape( in TEST_F()
923 auto inferred_status = ShapeInference::InferBinaryOpShape( in TEST_F()
1197 ShapeInference::InferBinaryOpShape(HloOpcode::kAdd, mat, vec8, {1}); in TEST_F()
1202 ShapeInference::InferBinaryOpShape(HloOpcode::kAdd, mat, vec8, {0}); in TEST_F()
1206 ShapeInference::InferBinaryOpShape(HloOpcode::kAdd, mat, vec16, {0}); in TEST_F()
1211 ShapeInference::InferBinaryOpShape(HloOpcode::kAdd, mat, vec16, {1}); in TEST_F()
1222 auto inferred_status_match = ShapeInference::InferBinaryOpShape( in TEST_F()
1227 inferred_status_match = ShapeInference::InferBinaryOpShape( in TEST_F()
1232 inferred_status_match = ShapeInference::InferBinaryOpShape( in TEST_F()
[all …]
|
| D | shape_inference.h | 56 static StatusOr<Shape> InferBinaryOpShape(
59 static StatusOr<Shape> InferBinaryOpShape(HloOpcode opcode,
|
| D | hlo_creation_utils.cc | 40 ShapeInference::InferBinaryOpShape(opcode, lhs, rhs)); in MakeBinaryHlo()
52 ShapeInference::InferBinaryOpShape(HloOpcode::kCompare, lhs, rhs)); in MakeCompareHlo()
|
| D | shape_inference.cc | 909 /* static */ StatusOr<Shape> ShapeInference::InferBinaryOpShape( in InferBinaryOpShape() function in xla::ShapeInference
911 return InferBinaryOpShape(opcode, lhs->shape(), rhs->shape(), in InferBinaryOpShape()
915 /* static */ StatusOr<Shape> ShapeInference::InferBinaryOpShape( in InferBinaryOpShape() function in xla::ShapeInference
|
| D | hlo_verifier.cc | 924 instruction, ShapeInference::InferBinaryOpShape(instruction->opcode(), in CheckBinaryShape()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 490 ShapeInference::InferBinaryOpShape( in BinaryOp()
|