Searched refs:InferAllGatherShape (Results 1 – 4 of 4) sorted by relevance
130 static StatusOr<Shape> InferAllGatherShape(const Shape& operand_shape,
275 return CheckShape(ag, ShapeInference::InferAllGatherShape( in HandleAllGather()
2034 /* static */ StatusOr<Shape> ShapeInference::InferAllGatherShape( in InferAllGatherShape() function in xla::ShapeInference
2725 ShapeInference::InferAllGatherShape( in AllGather()