Searched refs:InferCollectivePermuteDoneShape (Results 1 – 3 of 3) sorted by relevance
181 static StatusOr<Shape> InferCollectivePermuteDoneShape(
734 return CheckShape(hlo, ShapeInference::InferCollectivePermuteDoneShape( in HandleCollectivePermuteDone()
2210 /* static */ StatusOr<Shape> ShapeInference::InferCollectivePermuteDoneShape( in InferCollectivePermuteDoneShape() function in xla::ShapeInference