Home
last modified time | relevance | path

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

12

/external/llvm-project/mlir/include/mlir/IR/
DAffineExprVisitor.h107 expr.cast<AffineDimExpr>()); in walkPostOrder()
144 expr.cast<AffineDimExpr>()); in visit()
179 void visitDimExpr(AffineDimExpr expr) {} in visitDimExpr()
278 void visitDimExpr(AffineDimExpr expr);
DAffineExpr.h195 class AffineDimExpr : public AffineExpr {
198 /* implicit */ AffineDimExpr(AffineExpr::ImplType *ptr);
251 if (std::is_same<U, AffineDimExpr>::value) in isa()
/external/llvm-project/mlir/lib/IR/
DAffineMap.cpp61 if (auto attr = operandConsts[expr.cast<AffineDimExpr>().getPosition()] in constantFoldImpl()
133 if (auto d = e.dyn_cast<AffineDimExpr>()) in getMaxDimAndSymbol()
184 auto expr = results[i].dyn_cast<AffineDimExpr>(); in isIdentity()
231 return getResult(idx).cast<AffineDimExpr>().getPosition(); in getDimPosition()
356 if (auto dim = expr.dyn_cast<AffineDimExpr>()) { in isProjectedPermutation()
425 if (auto d = expr.dyn_cast<AffineDimExpr>()) { in inversePermutation()
DAffineExpr.cpp35 void visitDimExpr(AffineDimExpr expr) { callback(expr); } in walk()
68 unsigned dimId = cast<AffineDimExpr>().getPosition(); in replaceDimsAndSymbols()
273 AffineDimExpr::AffineDimExpr(AffineExpr::ImplType *ptr) : AffineExpr(ptr) {} in AffineDimExpr() function in AffineDimExpr
274 unsigned AffineDimExpr::getPosition() const { in getPosition()
960 void SimpleAffineExprFlattener::visitDimExpr(AffineDimExpr expr) { in visitDimExpr()
DBuiltinTypes.cpp506 if (auto dim = e.dyn_cast<AffineDimExpr>()) in extractStridesFromTerm()
533 auto dim = bin.getLHS().dyn_cast<AffineDimExpr>(); in extractStrides()
DAsmPrinter.cpp1997 unsigned pos = expr.cast<AffineDimExpr>().getPosition(); in printAffineExprInternal()
/external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
DFusion.cpp71 unsigned loopPos = e.cast<AffineDimExpr>().getPosition(); in inferShapeComponents()
192 auto dimExpr = en2.value().dyn_cast<AffineDimExpr>(); in getShapeDefiningLoopRange()
195 if (loopDepth == en2.value().cast<AffineDimExpr>().getPosition()) { in getShapeDefiningLoopRange()
270 unsigned posInProducerLoop = en.value().cast<AffineDimExpr>().getPosition(); in fuse()
590 return expr.cast<AffineDimExpr>().getPosition(); in doesTransposeAccess()
706 unsigned position = expr.cast<AffineDimExpr>().getPosition(); in collectFusableLoops()
DFusionOnTensors.cpp301 collapsedDims.front().cast<AffineDimExpr>().getPosition(); in linearizeCollapsedDims()
465 unsigned pos = resultExpr.value().cast<AffineDimExpr>().getPosition(); in fuseWithReshapeByExpansion()
515 unsigned pos = expr.cast<AffineDimExpr>().getPosition(); in fuseWithReshapeByExpansion()
536 unsigned origDim = expr.cast<AffineDimExpr>().getPosition(); in fuseWithReshapeByExpansion()
DDropUnitDims.cpp197 if (AffineDimExpr dimExpr = expr.value().dyn_cast<AffineDimExpr>()) in matchAndRewrite()
DBufferize.cpp93 int64_t loopIndex = expr.cast<AffineDimExpr>().getPosition(); in allocateBuffersForResults()
DTiling.cpp97 void visitDimExpr(AffineDimExpr expr) { in visitDimExpr()
DLoops.cpp613 if (expr.dyn_cast<AffineDimExpr>() || expr.dyn_cast<AffineSymbolExpr>()) { in matchAndRewrite()
/external/llvm-project/mlir/include/mlir/Dialect/Affine/IR/
DAffineOps.h30 class AffineDimExpr; variable
478 AffineDimExpr renumberOneDim(Value v);
/external/llvm-project/mlir/include/mlir/Dialect/Linalg/IR/
DLinalgTraits.h149 unsigned pos = dim.cast<AffineDimExpr>().getPosition(); in verifyTrait()
/external/llvm-project/mlir/lib/Dialect/Linalg/EDSC/
DBuilders.cpp226 unsigned numDims = c.cast<AffineDimExpr>().getPosition() + 1; in linalg_generic_conv_nhwc()
263 unsigned numDims = kw.cast<AffineDimExpr>().getPosition() + 1; in linalg_generic_dilated_conv_nhwc()
/external/llvm-project/mlir/lib/CAPI/IR/
DAffineExpr.cpp64 return unwrap(affineExpr).cast<AffineDimExpr>().getPosition(); in mlirAffineDimExprGetPosition()
/external/llvm-project/mlir/lib/Analysis/
DLoopAnalysis.cpp245 if (auto dimExpr = expr.dyn_cast<AffineDimExpr>()) in isContiguousAccess()
/external/llvm-project/mlir/lib/Conversion/VectorToSCF/
DVectorToSCF.cpp432 auto dim = en.value().template dyn_cast<AffineDimExpr>(); in computeCoalescedIndex()
490 auto dim = expr.dyn_cast<AffineDimExpr>(); in emitWithBoundsChecks()
/external/llvm-project/mlir/lib/Dialect/Linalg/IR/
DLinalgOps.cpp80 if (auto d = result.dyn_cast<AffineDimExpr>()) { in createLoopRanges()
590 AffineDimExpr dimExpr = rhsExpr.cast<AffineDimExpr>(); in collapseReassociationMaps()
683 auto d = e.dyn_cast<AffineDimExpr>(); in isReassociationValid()
810 unsigned maxDim = getMaxPosOfType<AffineDimExpr>(reassociation); in getSymbolLessAffineMaps()
/external/llvm-project/mlir/lib/Dialect/Vector/
DVectorTransforms.cpp577 auto expr = results[i].dyn_cast<AffineDimExpr>(); in isIdentitySuffix()
2505 auto affinExp = exp.dyn_cast<AffineDimExpr>(); in distributPointwiseVectorOp()
2540 unsigned pos = expr.cast<AffineDimExpr>().getPosition(); in matchAndRewrite()
2576 unsigned pos = expr.cast<AffineDimExpr>().getPosition(); in matchAndRewrite()
DVectorOps.cpp504 auto dim = map.getResult(i).cast<AffineDimExpr>(); in getIterationIndexMap()
710 auto dim = e.dyn_cast<AffineDimExpr>(); in foldExtractOpFromTranspose()
1012 auto dim = expr.cast<AffineDimExpr>(); in build()
1869 auto dim = expr.dyn_cast<AffineDimExpr>(); in verifyPermutationMap()
/external/llvm-project/mlir/lib/Conversion/AffineToStandard/
DAffineToStandard.cpp190 Value visitDimExpr(AffineDimExpr expr) { in visitDimExpr()
/external/llvm-project/mlir/lib/Dialect/SDBM/
DSDBMExpr.cpp510 SDBMExpr visitDimExpr(AffineDimExpr expr) { in tryConvertAffineExpr()
/external/llvm-project/mlir/lib/Dialect/Affine/IR/
DAffineOps.cpp437 if (auto dim = expr.dyn_cast<AffineDimExpr>()) in fold()
449 AffineDimExpr AffineApplyNormalizer::renumberOneDim(Value v) { in renumberOneDim()
458 .cast<AffineDimExpr>(); in renumberOneDim()
797 if (auto dimExpr = expr.dyn_cast<AffineDimExpr>()) in canonicalizeMapOrSetAndOperands()
/external/llvm-project/mlir/tools/mlir-linalg-ods-gen/
Dmlir-linalg-ods-gen.cpp1206 reductionDims.push_back(iter.cast<AffineDimExpr>().getPosition()); in parseExpression()

12