Lines Matching refs:reductionDimPos
246 unsigned reductionDimPos, in scaleReductionDim() argument
248 auto reductionDim = getAffineDimExpr(reductionDimPos, op.getContext()); in scaleReductionDim()
249 auto reductionDimP1 = getAffineDimExpr(reductionDimPos + 1, op.getContext()); in scaleReductionDim()
253 AffineMap shiftedIdMap = idMap.shiftDims(1, /*offset=*/reductionDimPos + 1); in scaleReductionDim()
261 unsigned reductionDimPos, int64_t size) { in insertParallelDim() argument
262 auto reductionDim = getAffineDimExpr(reductionDimPos, op.getContext()); in insertParallelDim()
266 AffineMap shiftedIdMap = idMap.shiftDims(1, /*offset=*/reductionDimPos + 1); in insertParallelDim()
267 return map.compose(shiftedIdMap).insertResult(reductionDim, reductionDimPos); in insertParallelDim()
289 unsigned reductionDimPos = dims[0]; in splitReductionByScaling() local
291 int64_t reductionDimSize = loopRanges[reductionDimPos]; in splitReductionByScaling()
361 newMaps.push_back(scaleReductionDim(op, *o, reductionDimPos, splitFactor)); in splitReductionByScaling()
364 auto redDim = getAffineDimExpr(reductionDimPos, context); in splitReductionByScaling()
365 auto redDimP1 = getAffineDimExpr(reductionDimPos + 1, context); in splitReductionByScaling()
372 newMaps.push_back(insertParallelDim(op, *o, reductionDimPos, in splitReductionByScaling()
389 iteratorTypes.insert(iteratorTypes.begin() + reductionDimPos, in splitReductionByScaling()
396 genericOp.region().front().insertArgument(reductionDimPos, in splitReductionByScaling()