Searched defs:resultSharding (Results 1 – 2 of 2) sorted by relevance
188 createAllReduceForResultWithoutPartialSharding(Value unshardedLinalgOpResult,ArrayRef<MeshAxis> opReductionMeshAxes,MeshShardingAttr resultSharding,ReductionKind reductionKind,IRMapping & spmdizationMap,ImplicitLocOpBuilder & builder) createAllReduceForResultWithoutPartialSharding() argument
102 MeshSharding resultSharding = MeshSharding::get( in handlePartialAxesDuringResharding() local