Home
last modified time | relevance | path

Searched refs:AffineDimExpr (Results 1 – 25 of 45) sorted by relevance

12

/llvm-project-15.0.7/mlir/test/python/ir/
H A Daffine_expr.py87 d11 = AffineDimExpr.get(1)
88 d2 = AffineDimExpr.get(2)
130 d1 = AffineDimExpr.get(1)
131 d2 = AffineDimExpr.get(2)
157 d1 = AffineDimExpr.get(1)
185 d1 = AffineDimExpr.get(1)
226 d1 = AffineDimExpr.get(1)
254 d1 = AffineDimExpr.get(1)
282 d1 = AffineDimExpr.get(1)
375 expr = AffineAddExpr.get(AffineDimExpr.get(0), AffineDimExpr.get(2))
[all …]
H A Daffine_map.py32 d0 = AffineDimExpr.get(0)
33 d1 = AffineDimExpr.get(1)
126 d0 = AffineDimExpr.get(0)
127 d1 = AffineDimExpr.get(1)
128 d2 = AffineDimExpr.get(2)
150 d0 = AffineDimExpr.get(0)
151 d1 = AffineDimExpr.get(1)
152 d2 = AffineDimExpr.get(2)
182 d0, d1, d2 = (AffineDimExpr.get(0), AffineDimExpr.get(1),
209 d0, d1, d2 = (AffineDimExpr.get(0), AffineDimExpr.get(1),
[all …]
H A Dinteger_set.py31 d0 = AffineDimExpr.get(0)
32 d1 = AffineDimExpr.get(1)
101 d0 = AffineDimExpr.get(0)
102 d1 = AffineDimExpr.get(1)
133 d0 = AffineDimExpr.get(0)
134 d1 = AffineDimExpr.get(1)
H A Dattributes.py149 d0 = AffineDimExpr.get(0)
150 d1 = AffineDimExpr.get(1)
/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/
H A DTensorInferTypeOpInterfaceImpl.cpp26 map.value().getResults().front().cast<AffineDimExpr>().getPosition(); in getExpandedDimToCollapsedDimMap()
28 map.value().getResults().back().cast<AffineDimExpr>().getPosition(); in getExpandedDimToCollapsedDimMap()
46 map.getResults().front().cast<AffineDimExpr>().getPosition(); in getCollapsedOutputDimFromInputShape()
47 unsigned endPos = map.getResults().back().cast<AffineDimExpr>().getPosition(); in getCollapsedOutputDimFromInputShape()
85 .cast<AffineDimExpr>() in getExpandedOutputDimFromInputShape()
90 .cast<AffineDimExpr>() in getExpandedOutputDimFromInputShape()
/llvm-project-15.0.7/mlir/include/mlir/IR/
H A DAffineExprVisitor.h109 expr.cast<AffineDimExpr>()); in walkPostOrder()
146 expr.cast<AffineDimExpr>()); in visit()
181 RetTy visitDimExpr(AffineDimExpr expr) { return RetTy(); } in visitDimExpr()
280 void visitDimExpr(AffineDimExpr expr);
H A DAffineExpr.h216 class AffineDimExpr : public AffineExpr {
219 /* implicit */ AffineDimExpr(AffineExpr::ImplType *ptr);
273 if (std::is_same<U, AffineDimExpr>::value) in isa()
H A DAffineMap.h554 if (auto dimExpr = expr.dyn_cast<AffineDimExpr>()) { in applyPermutationMap()
575 if (auto d = e.dyn_cast<AffineDimExpr>()) in getMaxDimAndSymbol()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/
H A DLinalgInterfaces.cpp201 LogicalResult visitDimExpr(AffineDimExpr dimExpr) { in visitDimExpr()
223 if (auto dimExpr = expr.dyn_cast<AffineDimExpr>()) { in isDimExprOrMulExpr()
242 auto dimExpr = getAffineExprOfType<AffineDimExpr>(lhsExpr, rhsExpr); in isDimExprOrMulExpr()
261 preservedDims.insert(expr.cast<AffineDimExpr>().getPosition()); in getPreservedDims()
320 unsigned outputDim = outputExpr.cast<AffineDimExpr>().getPosition(); in isConvolutionInterfaceImpl()
361 unsigned filterDim = filterExpr.cast<AffineDimExpr>().getPosition(); in isConvolutionInterfaceImpl()
517 if (auto d = result.dyn_cast<AffineDimExpr>()) { in createLoopRanges()
533 if (auto d = result.dyn_cast<AffineDimExpr>()) in computeStaticLoopSizes()
550 bool visitDimExpr(AffineDimExpr dimExpr) { in visitDimExpr()
795 if (indexingMap.getResult(dim).dyn_cast<AffineDimExpr>()) { in verifyStructuredOpInterface()
/llvm-project-15.0.7/mlir/lib/IR/
H A DAffineExpr.cpp37 void visitDimExpr(AffineDimExpr expr) { callback(expr); } in walk()
70 unsigned dimId = cast<AffineDimExpr>().getPosition(); in replaceDimsAndSymbols()
311 AffineDimExpr::AffineDimExpr(AffineExpr::ImplType *ptr) : AffineExpr(ptr) {} in AffineDimExpr() function in AffineDimExpr
312 unsigned AffineDimExpr::getPosition() const { in getPosition()
1028 if (!((lhs.isa<AffineDimExpr>() || lhs.isa<AffineSymbolExpr>()) && in getSemiAffineExprFromFlatForm()
1029 (rhs.isa<AffineDimExpr>() || rhs.isa<AffineSymbolExpr>() || in getSemiAffineExprFromFlatForm()
1038 if (lhs.isa<AffineDimExpr>()) { in getSemiAffineExprFromFlatForm()
1039 lhsPos = lhs.cast<AffineDimExpr>().getPosition(); in getSemiAffineExprFromFlatForm()
1050 } else if (lhs.isa<AffineDimExpr>()) { in getSemiAffineExprFromFlatForm()
1056 lhsPos = lhs.cast<AffineDimExpr>().getPosition(); in getSemiAffineExprFromFlatForm()
[all …]
H A DAffineMap.cpp62 if (auto attr = operandConsts[expr.cast<AffineDimExpr>().getPosition()] in constantFoldImpl()
133 } else if (auto dimExpr = expr.dyn_cast<AffineDimExpr>()) { in isMinorIdentityWithBroadcasting()
180 } else if (auto dimExpr = expr.dyn_cast<AffineDimExpr>()) { in isPermutationOfMinorIdentityWithBroadcasting()
260 auto expr = results[i].dyn_cast<AffineDimExpr>(); in isIdentity()
316 return getResult(idx).cast<AffineDimExpr>().getPosition(); in getDimPosition()
493 if (auto dim = expr.dyn_cast<AffineDimExpr>()) { in isProjectedPermutation()
661 if (auto d = expr.dyn_cast<AffineDimExpr>()) { in inversePermutation()
/llvm-project-15.0.7/mlir/lib/Dialect/Utils/
H A DReshapeOpsUtils.cpp176 indices.push_back(expr.cast<AffineDimExpr>().getPosition()); in convertReassociationMapsToIndices()
184 unsigned maxDim = getMaxPosOfType<AffineDimExpr>(reassociation); in getSymbolLessAffineMaps()
210 auto d = e.dyn_cast<AffineDimExpr>(); in isReassociationValid()
/llvm-project-15.0.7/mlir/lib/CAPI/IR/
H A DAffineExpr.cpp69 return unwrap(affineExpr).isa<AffineDimExpr>(); in mlirAffineExprIsADim()
77 return unwrap(affineExpr).cast<AffineDimExpr>().getPosition(); in mlirAffineDimExprGetPosition()
/llvm-project-15.0.7/mlir/lib/Interfaces/
H A DVectorInterfaces.cpp19 if (map.getResult(i).template isa<AffineDimExpr>()) { in transferMaskType()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DElementwiseOpFusion.cpp110 if (auto dimExpr = result.dyn_cast<AffineDimExpr>()) in areElementwiseOpsFusable()
531 unsigned pos = resultExpr.value().cast<AffineDimExpr>().getPosition(); in compute()
589 unsigned pos = expr.cast<AffineDimExpr>().getPosition(); in getIndexingMapInExpandedOp()
608 unsigned dim = expr.cast<AffineDimExpr>().getPosition(); in getExpandedType()
627 unsigned dim = expr.cast<AffineDimExpr>().getPosition(); in getReassociationForExpansion()
906 .cast<AffineDimExpr>() in getDomainReassociation()
929 unsigned dimInMapStart = expr.value().cast<AffineDimExpr>().getPosition(); in isDimSequencePreserved()
938 .cast<AffineDimExpr>() in isDimSequencePreserved()
1228 unsigned dim = expr.cast<AffineDimExpr>().getPosition(); in getCollapsedOpIndexingMap()
1254 indexingMap.getResult(counter).cast<AffineDimExpr>().getPosition(); in getOperandReassociation()
H A DFusion.cpp91 auto dimExpr = en.value().dyn_cast<AffineDimExpr>(); in getShapeDefiningLoopRange()
94 if (loopDepth == en.value().cast<AffineDimExpr>().getPosition()) { in getShapeDefiningLoopRange()
199 unsigned posInProducerLoop = en.value().cast<AffineDimExpr>().getPosition(); in fuse()
H A DBubbleUpExtractSlice.cpp115 unsigned position = result.value().cast<AffineDimExpr>().getPosition(); in matchAndRewrite()
H A DDropUnitDims.cpp190 if (AffineDimExpr dimExpr = expr.value().dyn_cast<AffineDimExpr>()) in matchAndRewrite()
H A DTilingInterfaceImpl.cpp170 resultExpr.value().cast<AffineDimExpr>().getPosition(); in generateResultTileValue()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorTransferPermutationMapRewritePatterns.cpp98 if (auto expr = comp.getResult(i).dyn_cast<AffineDimExpr>()) in matchAndRewrite()
168 return expr.dyn_cast<AffineDimExpr>().getPosition(); in matchAndRewrite()
H A DVectorUnrollDistribute.cpp121 unsigned pos = dim.value().cast<AffineDimExpr>().getPosition(); in sliceTransferIndices()
660 auto indexExpr = std::get<0>(it).dyn_cast<AffineDimExpr>(); in matchAndRewrite()
664 unsigned vectorPos = std::get<1>(it).cast<AffineDimExpr>().getPosition(); in matchAndRewrite()
707 auto indexExpr = std::get<0>(it).dyn_cast<AffineDimExpr>(); in matchAndRewrite()
711 unsigned vectorPos = std::get<1>(it).cast<AffineDimExpr>().getPosition(); in matchAndRewrite()
H A DVectorDistribute.cpp371 auto indexExpr = std::get<0>(it).dyn_cast<AffineDimExpr>(); in tryDistributeOp()
375 unsigned vectorPos = std::get<1>(it).cast<AffineDimExpr>().getPosition(); in tryDistributeOp()
604 auto indexExpr = std::get<0>(it).dyn_cast<AffineDimExpr>(); in matchAndRewrite()
608 unsigned vectorPos = std::get<1>(it).cast<AffineDimExpr>().getPosition(); in matchAndRewrite()
/llvm-project-15.0.7/mlir/python/mlir/dialects/linalg/opdsl/lang/
H A Daffine.py247 return _ir.AffineDimExpr.get(position=pos)
/llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/
H A DSparsification.cpp164 unsigned idx = a.cast<AffineDimExpr>().getPosition(); in findAffine()
233 unsigned idx = a.cast<AffineDimExpr>().getPosition(); in addAffineOrderings()
513 unsigned idx = a.cast<AffineDimExpr>().getPosition(); in genBuffers()
659 unsigned idx = a.cast<AffineDimExpr>().getPosition(); in genAffine()
689 unsigned idx = a.cast<AffineDimExpr>().getPosition(); in genIndex()
706 unsigned idx = a.cast<AffineDimExpr>().getPosition(); in genSubscript()
976 unsigned idx = a.cast<AffineDimExpr>().getPosition(); in isInvariantAffine()
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Analysis/
H A DLoopAnalysis.cpp233 if (auto dimExpr = expr.dyn_cast<AffineDimExpr>()) in isContiguousAccess()

12