Lines Matching refs:AffineMap
117 SmallVector<AffineMap> newMaps; in splitReduction()
120 AffineMap map = op.getTiedIndexingMap(operand); in splitReduction()
142 AffineMap::get(map.getNumDims() + 1, 0, exprs, op.getContext())); in splitReduction()
159 AffineMap oldOutputMap = op.getTiedIndexingMap(op.getOutputOperand(0)); in splitReduction()
191 newMaps.push_back(AffineMap::get(oldOutputMap.getNumDims() + 1, 0, outputExpr, in splitReduction()
210 AffineMap inputMap = b.getMultiDimIdentityMap(intermRank); in splitReduction()
222 AffineMap outputMap = AffineMap::get(intermRank, 0, exprs, op.getContext()); in splitReduction()
223 SmallVector<AffineMap> reductionMaps = {inputMap, outputMap}; in splitReduction()
245 static AffineMap scaleReductionDim(LinalgOp op, OpOperand &opOperand, in scaleReductionDim()
250 AffineMap map = op.getTiedIndexingMap(&opOperand); in scaleReductionDim()
251 AffineMap idMap = in scaleReductionDim()
252 AffineMap::getMultiDimIdentityMap(map.getNumDims(), op.getContext()); in scaleReductionDim()
253 AffineMap shiftedIdMap = idMap.shiftDims(1, /*offset=*/reductionDimPos + 1); in scaleReductionDim()
254 AffineMap composeMap = shiftedIdMap.replace( in scaleReductionDim()
260 static AffineMap insertParallelDim(LinalgOp op, OpOperand &opOperand, in insertParallelDim()
263 AffineMap map = op.getTiedIndexingMap(&opOperand); in insertParallelDim()
264 AffineMap idMap = in insertParallelDim()
265 AffineMap::getMultiDimIdentityMap(map.getNumDims(), op.getContext()); in insertParallelDim()
266 AffineMap shiftedIdMap = idMap.shiftDims(1, /*offset=*/reductionDimPos + 1); in insertParallelDim()
358 SmallVector<AffineMap> newMaps; in splitReductionByScaling()
366 newMaps.push_back(AffineMap::get(nDims, 0, {redDim, redDimP1}, context)); in splitReductionByScaling()
414 AffineMap map = b.getMultiDimIdentityMap(originalOutputType.getRank() + 1); in splitReductionByScaling()
415 SmallVector<AffineMap> indexingMaps = { in splitReductionByScaling()