Home
last modified time | relevance | path

Searched refs:symReplacements (Results 1 – 9 of 9) sorted by relevance

/llvm-project-15.0.7/mlir/lib/IR/
H A DIntegerSet.cpp69 ArrayRef<AffineExpr> dimReplacements, ArrayRef<AffineExpr> symReplacements, in replaceDimsAndSymbols() argument
75 cst.replaceDimsAndSymbols(dimReplacements, symReplacements)); in replaceDimsAndSymbols()
H A DAffineMap.cpp393 ArrayRef<AffineExpr> symReplacements, in replaceDimsAndSymbols() argument
400 expr.replaceDimsAndSymbols(dimReplacements, symReplacements)); in replaceDimsAndSymbols()
604 SmallVector<AffineExpr> symReplacements; in compressSymbols() local
605 symReplacements.reserve(map.getNumSymbols()); in compressSymbols()
609 symReplacements.push_back(getAffineConstantExpr(0, context)); in compressSymbols()
611 symReplacements.push_back(getAffineSymbolExpr(numSymbols++, context)); in compressSymbols()
616 resultExprs.push_back(e.replaceSymbols(symReplacements)); in compressSymbols()
H A DAffineExpr.cpp65 ArrayRef<AffineExpr> symReplacements) const { in replaceDimsAndSymbols()
77 if (symId >= symReplacements.size()) in replaceDimsAndSymbols()
79 return symReplacements[symId]; in replaceDimsAndSymbols()
88 auto newLHS = lhs.replaceDimsAndSymbols(dimReplacements, symReplacements); in replaceDimsAndSymbols()
89 auto newRHS = rhs.replaceDimsAndSymbols(dimReplacements, symReplacements); in replaceDimsAndSymbols()
102 AffineExpr::replaceSymbols(ArrayRef<AffineExpr> symReplacements) const { in replaceSymbols()
103 return replaceDimsAndSymbols({}, symReplacements); in replaceSymbols()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DDropUnitDims.cpp119 SmallVector<AffineExpr, 4> symReplacements; in replaceUnitDims() local
120 symReplacements.reserve(numSymbols); in replaceUnitDims()
122 symReplacements.push_back(getAffineSymbolExpr(symbol, context)); in replaceUnitDims()
131 operandMap.replaceDimsAndSymbols(dimReplacements, symReplacements, in replaceUnitDims()
/llvm-project-15.0.7/mlir/include/mlir/IR/
H A DAffineExpr.h130 ArrayRef<AffineExpr> symReplacements) const;
136 AffineExpr replaceSymbols(ArrayRef<AffineExpr> symReplacements) const;
H A DIntegerSet.h72 ArrayRef<AffineExpr> symReplacements,
H A DAffineMap.h197 ArrayRef<AffineExpr> symReplacements,
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Analysis/
H A DAffineStructures.cpp1582 SmallVector<AffineExpr> symReplacements(map.getNumSymbols(), {}); in alignAffineMapWithValues() local
1610 symReplacements[operand.index() - map.getNumDims()] = replacement; in alignAffineMapWithValues()
1614 return map.replaceDimsAndSymbols(dimReplacements, symReplacements, in alignAffineMapWithValues()
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/IR/
H A DAffineOps.cpp672 SmallVector<AffineExpr, 4> dimReplacements, symReplacements; in composeAffineMapAndOperands() local
674 symReplacements.reserve(syms.size()); in composeAffineMapAndOperands()
677 auto &repls = isDim ? dimReplacements : symReplacements; in composeAffineMapAndOperands()
692 *map = map->replaceDimsAndSymbols(dimReplacements, symReplacements, nDims, in composeAffineMapAndOperands()