Searched refs:InferCollectivePermuteStartShape (Results 1 – 4 of 4) sorted by relevance
133 ShapeInference::InferCollectivePermuteStartShape( in Run()
177 static StatusOr<Shape> InferCollectivePermuteStartShape(
730 hlo, ShapeInference::InferCollectivePermuteStartShape(operand_shapes)); in HandleCollectivePermuteStart()
2195 /* static */ StatusOr<Shape> ShapeInference::InferCollectivePermuteStartShape( in InferCollectivePermuteStartShape() function in xla::ShapeInference