Searched refs:shiftDims (Results 1 – 7 of 7) sorted by relevance
| /llvm-project-15.0.7/mlir/include/mlir/IR/ |
| H A D | AffineMap.h | 220 AffineMap shiftDims(unsigned shift, unsigned offset = 0) const {
226 return e.shiftDims(getNumDims(), shift, offset);
|
| H A D | AffineExpr.h | 149 AffineExpr shiftDims(unsigned numDims, unsigned shift,
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/IR/ |
| H A D | AffineValueMap.cpp | 43 .shiftDims(a.getNumDims()) in difference()
|
| H A D | AffineOps.cpp | 618 composeMap.shiftDims(dims.size()).shiftSymbols(syms.size()).getResult(0); in replaceDimOrSym()
835 submap = submap.shiftDims(dims.size()).shiftSymbols(symbols.size()); in composeMultiResultAffineMap()
2959 newExprs.push_back(expr.shiftDims(numProducerDims, numUsedDims) in matchAndRewrite()
3703 .shiftDims(numDims, totalNumDims) in parseAffineMapWithMinMax()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | SplitReduction.cpp | 253 AffineMap shiftedIdMap = idMap.shiftDims(1, /*offset=*/reductionDimPos + 1); in scaleReductionDim()
266 AffineMap shiftedIdMap = idMap.shiftDims(1, /*offset=*/reductionDimPos + 1); in insertParallelDim()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SCF/Utils/ |
| H A D | AffineCanonicalizationUtils.cpp | 138 AffineMap alignedBoundMap = boundMap.shiftDims(/*shift=*/1, /*offset=*/dimOp); in canonicalizeMinMaxOp()
|
| /llvm-project-15.0.7/mlir/lib/IR/ |
| H A D | AffineExpr.cpp | 108 AffineExpr AffineExpr::shiftDims(unsigned numDims, unsigned shift, in shiftDims() function in AffineExpr
|