| /llvm-project-15.0.7/mlir/lib/IR/ |
| H A D | IntegerSet.cpp | 69 ArrayRef<AffineExpr> dimReplacements, ArrayRef<AffineExpr> symReplacements, in replaceDimsAndSymbols() argument
75 cst.replaceDimsAndSymbols(dimReplacements, symReplacements)); in replaceDimsAndSymbols()
|
| H A D | AffineMap.cpp | 392 AffineMap AffineMap::replaceDimsAndSymbols(ArrayRef<AffineExpr> dimReplacements, in replaceDimsAndSymbols() argument
400 expr.replaceDimsAndSymbols(dimReplacements, symReplacements)); in replaceDimsAndSymbols()
546 SmallVector<AffineExpr> dimReplacements; in compressDims() local
547 dimReplacements.reserve(map.getNumDims()); in compressDims()
551 dimReplacements.push_back(getAffineConstantExpr(0, context)); in compressDims()
553 dimReplacements.push_back(getAffineDimExpr(numDims++, context)); in compressDims()
558 resultExprs.push_back(e.replaceDims(dimReplacements)); in compressDims()
|
| H A D | AffineExpr.cpp | 64 AffineExpr::replaceDimsAndSymbols(ArrayRef<AffineExpr> dimReplacements, in replaceDimsAndSymbols() argument
71 if (dimId >= dimReplacements.size()) in replaceDimsAndSymbols()
73 return dimReplacements[dimId]; in replaceDimsAndSymbols()
88 auto newLHS = lhs.replaceDimsAndSymbols(dimReplacements, symReplacements); in replaceDimsAndSymbols()
89 auto newRHS = rhs.replaceDimsAndSymbols(dimReplacements, symReplacements); in replaceDimsAndSymbols()
97 AffineExpr AffineExpr::replaceDims(ArrayRef<AffineExpr> dimReplacements) const { in replaceDims()
98 return replaceDimsAndSymbols(dimReplacements, {}); in replaceDims()
877 SmallVector<AffineExpr, 8> dimReplacements(map.getResults().begin(), in compose() local
879 return replaceDimsAndSymbols(dimReplacements, {}); in compose()
|
| /llvm-project-15.0.7/mlir/lib/CAPI/IR/ |
| H A D | IntegerSet.cpp | 57 const MlirAffineExpr *dimReplacements, in mlirIntegerSetReplaceGet() argument
61 (void)unwrapList(unwrap(set).getNumDims(), dimReplacements, mlirDims); in mlirIntegerSetReplaceGet()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | DropUnitDims.cpp | 108 SmallVector<AffineExpr, 4> dimReplacements; in replaceUnitDims() local
109 dimReplacements.reserve(numIterationDims); in replaceUnitDims()
113 dimReplacements.push_back(getAffineConstantExpr(0, context)); in replaceUnitDims()
115 dimReplacements.push_back(getAffineDimExpr(numKeptDims++, context)); in replaceUnitDims()
131 operandMap.replaceDimsAndSymbols(dimReplacements, symReplacements, in replaceUnitDims()
|
| /llvm-project-15.0.7/mlir/include/mlir-c/ |
| H A D | IntegerSet.h | 90 MlirIntegerSet set, const MlirAffineExpr *dimReplacements,
|
| /llvm-project-15.0.7/mlir/include/mlir/IR/ |
| H A D | AffineExpr.h | 129 AffineExpr replaceDimsAndSymbols(ArrayRef<AffineExpr> dimReplacements,
133 AffineExpr replaceDims(ArrayRef<AffineExpr> dimReplacements) const;
|
| H A D | IntegerSet.h | 71 IntegerSet replaceDimsAndSymbols(ArrayRef<AffineExpr> dimReplacements,
|
| H A D | AffineMap.h | 196 AffineMap replaceDimsAndSymbols(ArrayRef<AffineExpr> dimReplacements,
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/Analysis/ |
| H A D | AffineStructures.cpp | 1580 SmallVector<AffineExpr> dimReplacements(map.getNumDims(), {}); in alignAffineMapWithValues() local
1608 dimReplacements[operand.index()] = replacement; in alignAffineMapWithValues()
1614 return map.replaceDimsAndSymbols(dimReplacements, symReplacements, in alignAffineMapWithValues()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/IR/ |
| H A D | AffineOps.cpp | 672 SmallVector<AffineExpr, 4> dimReplacements, symReplacements; in composeAffineMapAndOperands() local
673 dimReplacements.reserve(dims.size()); in composeAffineMapAndOperands()
677 auto &repls = isDim ? dimReplacements : symReplacements; in composeAffineMapAndOperands()
692 *map = map->replaceDimsAndSymbols(dimReplacements, symReplacements, nDims, in composeAffineMapAndOperands()
|