Searched defs:reductionLoopKinds (Results 1 – 1 of 1) sorted by relevance
395 getShardingAttribute(OpResult result,const ShardingOption & shardingOption,AffineMap map,ArrayRef<utils::IteratorType> loopTypes,ArrayRef<ReductionKind> reductionLoopKinds) getShardingAttribute() argument 508 addShardOp(OpBuilder & b,OpResult result,const ShardingOption & shardingOption,AffineMap map,ArrayRef<utils::IteratorType> loopTypes,ArrayRef<ReductionKind> reductionLoopKinds) addShardOp() argument