Searched refs:dimExpr (Results 1 – 10 of 10) sorted by relevance
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | Transforms.cpp | 386 static AffineExpr substituteLoopInExpr(AffineExpr expr, AffineExpr dimExpr, in substituteLoopInExpr() argument
398 AffineExpr ee = substWithMin(expr, dimExpr, lb, in substituteLoopInExpr()
415 AffineExpr dimExpr = getAffineDimExpr(dimIdx, expr.getContext()); in substitute() local
416 LLVM_DEBUG(DBGS() << "Subst: " << dim << " @ " << dimExpr << "\n"); in substitute()
420 expr, dimExpr, forOp.lowerBound(), forOp.upperBound(), in substitute()
427 expr, dimExpr, parallelForOp.lowerBound()[idx], in substitute()
|
| D | DropUnitDims.cpp | 197 if (AffineDimExpr dimExpr = expr.value().dyn_cast<AffineDimExpr>()) in matchAndRewrite() local
198 if (dims[dimExpr.getPosition()] == 1 && in matchAndRewrite()
|
| D | Fusion.cpp | 192 auto dimExpr = en2.value().dyn_cast<AffineDimExpr>(); in getShapeDefiningLoopRange() local
193 if (!dimExpr) in getShapeDefiningLoopRange()
|
| /external/llvm-project/mlir/lib/Analysis/ |
| D | LoopAnalysis.cpp | 245 if (auto dimExpr = expr.dyn_cast<AffineDimExpr>()) in isContiguousAccess() local
246 exprOperands.push_back(mapOperands[dimExpr.getPosition()]); in isContiguousAccess()
|
| /external/llvm-project/mlir/lib/IR/ |
| D | BuiltinTypes.cpp | 766 AffineExpr dimExpr = std::get<0>(en); in makeCanonicalStridedLayoutExpr() local
770 expr = expr ? expr + dimExpr * stride : dimExpr * stride; in makeCanonicalStridedLayoutExpr()
|
| /external/llvm-project/mlir/lib/Dialect/SDBM/ |
| D | SDBM.cpp | 32 void visitDim(SDBMDimExpr dimExpr) { in collectSDBMBuildInfo()
33 int p = dimExpr.getPosition(); in collectSDBMBuildInfo()
|
| /external/llvm-project/mlir/lib/Transforms/ |
| D | LoopFusion.cpp | 854 auto dimExpr = b.getAffineDimExpr(outerIVs.size() + i); in createPrivateMemRef() local
857 simplifyAffineExpr(dimExpr - offsets[i], outerIVs.size() + rank, 0); in createPrivateMemRef()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/IR/ |
| D | LinalgOps.cpp | 590 AffineDimExpr dimExpr = rhsExpr.cast<AffineDimExpr>(); in collapseReassociationMaps() local
591 for (int i = 0, e = mapsProducer[dimExpr.getPosition()].getNumResults(); in collapseReassociationMaps()
|
| /external/llvm-project/mlir/lib/Dialect/Affine/IR/ |
| D | AffineOps.cpp | 797 if (auto dimExpr = expr.dyn_cast<AffineDimExpr>()) in canonicalizeMapOrSetAndOperands() local
798 usedDims[dimExpr.getPosition()] = true; in canonicalizeMapOrSetAndOperands()
|
| /external/llvm-project/mlir/lib/Transforms/Utils/ |
| D | LoopUtils.cpp | 2570 auto dimExpr = b.getAffineDimExpr(regionSymbols.size() + i); in generateCopy() local
2571 remapExprs.push_back(dimExpr - fastBufOffsets[i]); in generateCopy()
|