Searched defs:reductionKind (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 211 ReductionKind reductionKind = getReductionKindOfLinalgOp(unshardedOp); createAllReduceForResultsWithoutPartialShardings() local 291 mesh::ReductionKind reductionKind = getReductionKindOfLinalgOp(linalgOp); getReductionLoopIteratorKinds() local
1001 static arith::ConstantOp createInitialVector(arith::AtomicRMWKind reductionKind, in createInitialVector() argument 1281 static bool isNeutralElementConst(arith::AtomicRMWKind reductionKind, in isNeutralElementConst() argument