Searched defs:opReductionMeshAxes (Results 1 – 1 of 1) sorted by relevance
187 createAllReduceForResultWithoutPartialSharding(Value unshardedLinalgOpResult,ArrayRef<MeshAxis> opReductionMeshAxes,MeshShardingAttr resultSharding,ReductionKind reductionKind,IRMapping & spmdizationMap,ImplicitLocOpBuilder & builder) createAllReduceForResultWithoutPartialSharding() argument 208 createAllReduceForResultsWithoutPartialShardings(LinalgOp unshardedOp,ArrayRef<MeshAxis> opReductionMeshAxes,ArrayRef<MeshShardingAttr> resultShardings,IRMapping & spmdizationMap,ImplicitLocOpBuilder & builder) createAllReduceForResultsWithoutPartialShardings() argument