Searched refs:InferCollectivePermuteShape (Results 1 – 4 of 4) sorted by relevance
143 static StatusOr<Shape> InferCollectivePermuteShape(const Shape& shape);
209 return CheckShape(hlo, ShapeInference::InferCollectivePermuteShape( in HandleCollectivePermute()
2036 /* static */ StatusOr<Shape> ShapeInference::InferCollectivePermuteShape( in InferCollectivePermuteShape() function in xla::ShapeInference
2222 ShapeInference::InferCollectivePermuteShape(operand_shape)); in CollectivePermute()