Searched refs:InferTransposeShape (Results 1 – 6 of 6) sorted by relevance
235 static StatusOr<Shape> InferTransposeShape(
102 ShapeInference::InferTransposeShape(operand->shape(), dimensions)); in MakeTransposeHlo()
505 transpose, ShapeInference::InferTransposeShape( in HandleTranspose()
1523 ShapeInference::InferTransposeShape(a_shape, {1, 2, 3, 0}); in TEST_F()1533 ShapeInference::InferTransposeShape(a_shape, {0}); in TEST_F()
2718 /* static */ StatusOr<Shape> ShapeInference::InferTransposeShape( in InferTransposeShape() function in xla::ShapeInference
1542 TF_ASSIGN_OR_RETURN(Shape shape, ShapeInference::InferTransposeShape( in Transpose()