Home
last modified time | relevance | path

Searched defs:dimExpr (Results 1 – 21 of 21) sorted by relevance

/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DDataLayoutPropagation.cpp132 if (auto dimExpr = dyn_cast<AffineDimExpr>(permutedExpr)) { in getPackingInfoFromOperand() local
183 if (auto dimExpr = dyn_cast<AffineDimExpr>(expr)) in computeOuterDims() local
244 if (auto dimExpr = dyn_cast<AffineDimExpr>(expr)) { in getOrCreatePackedViewOfOperand() local
270 if (auto dimExpr = dyn_cast<AffineDimExpr>(exprs[i])) { in getOrCreatePackedViewOfOperand() local
H A DFusion.cpp90 auto dimExpr = dyn_cast<AffineDimExpr>(en.value()); in getShapeDefiningLoopRange() local
H A DDropUnitDims.cpp352 if (auto dimExpr = dyn_cast<AffineDimExpr>(exprs[dim])) { dropUnitExtentFromOperandMetadata() local
415 if (AffineDimExpr dimExpr = dyn_cast<AffineDimExpr>(expr.value())) { dropUnitDims() local
H A DTilingInterfaceImpl.cpp419 for (AffineExpr dimExpr : valueMap.getResults()) { tileToPartialReduction() local
H A DElementwiseOpFusion.cpp143 if (auto dimExpr = dyn_cast<AffineDimExpr>(result)) areElementwiseOpsFusable() local
H A DVectorization.cpp604 __anon729377360e02(AffineDimExpr dimExpr) buildVectorWrite() argument
/llvm-project/mlir/include/mlir/IR/
H A DAffineMap.h655 if (auto dimExpr = dyn_cast<AffineDimExpr>(expr)) { applyPermutationMap() local
/llvm-project/mlir/lib/Dialect/SparseTensor/IR/Detail/
H A DVar.h161 DimVar(AffineDimExpr dimExpr) : Var(Kind, dimExpr) {} in DimVar()
/llvm-project/mlir/lib/Dialect/Affine/Analysis/
H A DLoopAnalysis.cpp212 if (auto dimExpr = dyn_cast<AffineDimExpr>(expr)) isContiguousAccess() local
/llvm-project/mlir/lib/Dialect/Vector/Transforms/
H A DLowerVectorTransfer.cpp343 auto dimExpr = dyn_cast<AffineConstantExpr>(expr); matchAndRewriteMaskableOp() local
H A DVectorDistribute.cpp100 for (auto dimExpr : distributionMap.getResults()) { buildStore() local
145 for (auto dimExpr : distributionMap.getResults()) { buildLoad() local
/llvm-project/mlir/lib/Dialect/Linalg/IR/
H A DLinalgInterfaces.cpp563 if (auto dimExpr = dyn_cast<AffineDimExpr>(expr)) { getDimExprOrMulExprDimPos() local
585 auto dimExpr = getAffineExprOfType<AffineDimExpr>(lhsExpr, rhsExpr); getDimExprOrMulExprDimPos() local
H A DLinalgOps.cpp2348 AffineExpr dimExpr = sourceMap.getResult(i); createNewOperandWithStaticSizes() local
/llvm-project/mlir/lib/IR/
H A DAffineMap.cpp179 } else if (auto dimExpr = dyn_cast<AffineDimExpr>(expr)) { isMinorIdentityWithBroadcasting() local
226 } else if (auto dimExpr = dyn_cast<AffineDimExpr>(expr)) { isPermutationOfMinorIdentityWithBroadcasting() local
H A DBuiltinTypes.cpp933 AffineExpr dimExpr = std::get<0>(en); makeCanonicalStridedLayoutExpr() local
/llvm-project/mlir/lib/Dialect/Mesh/Interfaces/
H A DShardingInterface.cpp53 AffineExpr dimExpr; in checkOperandAffineExprRecursively() local
/llvm-project/mlir/lib/Dialect/Affine/Transforms/
H A DLoopFusion.cpp324 auto dimExpr = b.getAffineDimExpr(outerIVs.size() + i); in createPrivateMemRef() local
/llvm-project/mlir/lib/Dialect/Affine/IR/
H A DAffineOps.cpp632 auto dimExpr = dyn_cast<AffineDimExpr>(e); getLargestKnownDivisor() local
666 auto dimExpr = dyn_cast<AffineDimExpr>(e); isNonNegativeBoundedBy() local
1374 if (auto dimExpr = dyn_cast<AffineDimExpr>(expr)) canonicalizeMapOrSetAndOperands() local
3359 } else if (auto dimExpr = dyn_cast<AffineDimExpr>(expr)) { matchAndRewrite() local
[all...]
/llvm-project/mlir/lib/Analysis/
H A DFlatLinearValueConstraints.cpp381 auto dimExpr = dyn_cast<AffineDimExpr>(dividendExpr); in detectAsMod() local
/llvm-project/mlir/lib/Dialect/Affine/Utils/
H A DLoopUtils.cpp2195 auto dimExpr = b.getAffineDimExpr(regionSymbols.size() + i); generateCopy() local
/llvm-project/mlir/lib/Dialect/Vector/IR/
H A DVectorOps.cpp4154 auto dimExpr = dyn_cast<AffineDimExpr>(permutationMap.getResult(i)); foldTransferInBoundsAttribute() local