Home
last modified time | relevance | path

Searched refs:InferDotOpShape (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference_test.cc1471 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 …]
Doperand_upcaster.cc27 return ShapeInference::InferDotOpShape( in MaybeInferShape()
Dshape_inference.h305 static StatusOr<Shape> InferDotOpShape(
Dhlo_creation_utils.cc295 ShapeInference::InferDotOpShape(lhs->shape(), rhs->shape(), dim_numbers, in MakeDotHlo()
Dhlo_verifier.cc143 ShapeInference::InferDotOpShape( in HandleDot()
Dhlo_evaluator.cc392 ShapeInference::InferDotOpShape( in EvaluateDotOp()
Dshape_inference.cc648 /* static */ StatusOr<Shape> ShapeInference::InferDotOpShape( in InferDotOpShape() function in xla::ShapeInference
Dhlo_parser.cc2429 return ShapeInference::InferDotOpShape( in ParseInstructionRhs()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Ddot_handler.cc83 ShapeInference::InferDotOpShape( in HandleDot()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc1308 ShapeInference::InferDotOpShape( in DotGeneral()