Searched refs:InferAllToAllTupleShape (Results 1 – 4 of 4) sorted by relevance
147 static StatusOr<Shape> InferAllToAllTupleShape(
331 ShapeInference::InferAllToAllTupleShape(operand_shapes)); in HandleAllToAll()
2088 /* static */ StatusOr<Shape> ShapeInference::InferAllToAllTupleShape( in InferAllToAllTupleShape() function in xla::ShapeInference
2960 Shape shape, ShapeInference::InferAllToAllTupleShape(slice_shape_ptrs)); in AllToAllTuple()