Searched refs:InferAllToAllTupleShape (Results 1 – 4 of 4) sorted by relevance
139 static StatusOr<Shape> InferAllToAllTupleShape(
201 ShapeInference::InferAllToAllTupleShape(operand_shapes)); in HandleAllToAll()
2018 /* static */ StatusOr<Shape> ShapeInference::InferAllToAllTupleShape( in InferAllToAllTupleShape() function in xla::ShapeInference
2195 Shape shape, ShapeInference::InferAllToAllTupleShape(slice_shape_ptrs)); in AllToAll()