Searched refs:InferDotOpShape (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference_test.cc | 969 ShapeInference::InferDotOpShape(f32_, vector_32_, dot_dnums); in TEST_F()
980 auto inferred_status = ShapeInference::InferDotOpShape( in TEST_F()
993 ShapeInference::InferDotOpShape(vector_64_, vector_64_, dot_dnums); in TEST_F()
997 ShapeInference::InferDotOpShape(vector_64_, vector_32_, dot_dnums); in TEST_F()
1007 ShapeInference::InferDotOpShape(matrix_32_64_, vector_64_, dot_dnums); in TEST_F()
1011 ShapeInference::InferDotOpShape(matrix_32_64_, vector_32_, dot_dnums); in TEST_F()
1021 ShapeInference::InferDotOpShape(vector_32_, matrix_32_64_, dot_dnums); in TEST_F()
1025 ShapeInference::InferDotOpShape(vector_64_, matrix_32_64_, dot_dnums); in TEST_F()
1035 ShapeInference::InferDotOpShape(matrix_32_64_, matrix_64_48_, dot_dnums); in TEST_F()
1043 ShapeInference::InferDotOpShape(matrix_32_64_, matrix_32_64_, dot_dnums); in TEST_F()
[all …]
|
| D | shape_inference.h | 275 static StatusOr<Shape> InferDotOpShape(
|
| D | hlo_creation_utils.cc | 224 ShapeInference::InferDotOpShape(lhs->shape(), rhs->shape(), dim_numbers)); in MakeDotHlo()
|
| D | hlo_verifier.cc | 148 ShapeInference::InferDotOpShape( in HandleDot()
|
| D | hlo_evaluator.cc | 374 ShapeInference::InferDotOpShape(lhs.shape(), rhs.shape(), dim_numbers)); in EvaluateDotOp()
|
| D | shape_inference.cc | 608 /* static */ StatusOr<Shape> ShapeInference::InferDotOpShape( in InferDotOpShape() function in xla::ShapeInference
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1043 ShapeInference::InferDotOpShape(lhs_shape, rhs_shape, in DotGeneral()
|