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