Searched refs:InferDotOpShape (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference_test.cc | 1471 auto inferred_status = ShapeInference::InferDotOpShape( in TEST_F()
1482 auto inferred_status = ShapeInference::InferDotOpShape( in TEST_F()
1496 ShapeInference::InferDotOpShape(vector_64_, vector_64_, dot_dnums, in TEST_F()
1501 ShapeInference::InferDotOpShape(vector_64_, vector_32_, dot_dnums, in TEST_F()
1512 ShapeInference::InferDotOpShape(matrix_32_64_, vector_64_, dot_dnums, in TEST_F()
1517 ShapeInference::InferDotOpShape(matrix_32_64_, vector_32_, dot_dnums, in TEST_F()
1528 ShapeInference::InferDotOpShape(vector_32_, matrix_32_64_, dot_dnums, in TEST_F()
1533 ShapeInference::InferDotOpShape(vector_64_, matrix_32_64_, dot_dnums, in TEST_F()
1544 ShapeInference::InferDotOpShape(matrix_32_64_, matrix_64_48_, dot_dnums, in TEST_F()
1553 ShapeInference::InferDotOpShape(matrix_32_64_, matrix_32_64_, dot_dnums, in TEST_F()
[all …]
|
| D | operand_upcaster.cc | 27 return ShapeInference::InferDotOpShape( in MaybeInferShape()
|
| D | shape_inference.h | 305 static StatusOr<Shape> InferDotOpShape(
|
| D | hlo_creation_utils.cc | 295 ShapeInference::InferDotOpShape(lhs->shape(), rhs->shape(), dim_numbers, in MakeDotHlo()
|
| D | hlo_verifier.cc | 143 ShapeInference::InferDotOpShape( in HandleDot()
|
| D | hlo_evaluator.cc | 392 ShapeInference::InferDotOpShape( in EvaluateDotOp()
|
| D | shape_inference.cc | 648 /* static */ StatusOr<Shape> ShapeInference::InferDotOpShape( in InferDotOpShape() function in xla::ShapeInference
|
| D | hlo_parser.cc | 2429 return ShapeInference::InferDotOpShape( in ParseInstructionRhs()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 83 ShapeInference::InferDotOpShape( in HandleDot()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1308 ShapeInference::InferDotOpShape( in DotGeneral()
|