Home
last modified time | relevance | path

Searched refs:dimReplacements (Results 1 – 7 of 7) sorted by relevance

/external/llvm-project/mlir/lib/IR/
DIntegerSet.cpp72 ArrayRef<AffineExpr> dimReplacements, ArrayRef<AffineExpr> symReplacements, in replaceDimsAndSymbols() argument
78 cst.replaceDimsAndSymbols(dimReplacements, symReplacements)); in replaceDimsAndSymbols()
DAffineExpr.cpp62 AffineExpr::replaceDimsAndSymbols(ArrayRef<AffineExpr> dimReplacements, in replaceDimsAndSymbols() argument
69 if (dimId >= dimReplacements.size()) in replaceDimsAndSymbols()
71 return dimReplacements[dimId]; in replaceDimsAndSymbols()
86 auto newLHS = lhs.replaceDimsAndSymbols(dimReplacements, symReplacements); in replaceDimsAndSymbols()
87 auto newRHS = rhs.replaceDimsAndSymbols(dimReplacements, symReplacements); in replaceDimsAndSymbols()
812 SmallVector<AffineExpr, 8> dimReplacements(map.getResults().begin(), in compose() local
814 return replaceDimsAndSymbols(dimReplacements, {}); in compose()
DAffineMap.cpp299 AffineMap AffineMap::replaceDimsAndSymbols(ArrayRef<AffineExpr> dimReplacements, in replaceDimsAndSymbols() argument
307 expr.replaceDimsAndSymbols(dimReplacements, symReplacements)); in replaceDimsAndSymbols()
/external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
DDropUnitDims.cpp108 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()
/external/llvm-project/mlir/include/mlir/IR/
DIntegerSet.h69 IntegerSet replaceDimsAndSymbols(ArrayRef<AffineExpr> dimReplacements,
DAffineMap.h141 AffineMap replaceDimsAndSymbols(ArrayRef<AffineExpr> dimReplacements,
DAffineExpr.h120 AffineExpr replaceDimsAndSymbols(ArrayRef<AffineExpr> dimReplacements,