Searched refs:all_gather_start_shape (Results 1 – 2 of 2) sorted by relevance
141 const Shape& all_gather_start_shape);
2075 const Shape& all_gather_start_shape) { in InferAllGatherDoneShape() argument2076 return ShapeUtil::GetTupleElementShape(all_gather_start_shape, 1); in InferAllGatherDoneShape()