Home
last modified time | relevance | path

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

12

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DSplitReduction.cpp131 exprs.push_back(b.getAffineDimExpr(insertSplitDimension)); in splitReduction()
133 b.getAffineDimExpr(dim < insertSplitDimension ? dim : dim + 1)); in splitReduction()
138 b.getAffineDimExpr(dim < insertSplitDimension ? dim : dim + 1)); in splitReduction()
166 outputExpr.push_back(b.getAffineDimExpr(insertSplitDimension)); in splitReduction()
173 b.getAffineDimExpr(dim < insertSplitDimension ? dim : dim + 1)); in splitReduction()
218 exprs.push_back(b.getAffineDimExpr(i)); in splitReduction()
248 auto reductionDim = getAffineDimExpr(reductionDimPos, op.getContext()); in scaleReductionDim()
249 auto reductionDimP1 = getAffineDimExpr(reductionDimPos + 1, op.getContext()); in scaleReductionDim()
262 auto reductionDim = getAffineDimExpr(reductionDimPos, op.getContext()); in insertParallelDim()
364 auto redDim = getAffineDimExpr(reductionDimPos, context); in splitReductionByScaling()
[all …]
H A DDropUnitDims.cpp115 dimReplacements.push_back(getAffineDimExpr(numKeptDims++, context)); in replaceUnitDims()
267 reassociations.push_back(getAffineDimExpr(dim++, context)); in replaceUnitExtents()
269 reassociations.push_back(getAffineDimExpr(dim, context)); in replaceUnitExtents()
275 reassociations.push_back(getAffineDimExpr(dim, context)); in replaceUnitExtents()
392 getAffineDimExpr(dim, context), context)); in matchAndRewrite()
H A DSplit.cpp134 AffineExpr d0 = rewriter.getAffineDimExpr(0); in splitOp()
135 AffineExpr d1 = rewriter.getAffineDimExpr(1); in splitOp()
/llvm-project-15.0.7/mlir/lib/Dialect/MemRef/Transforms/
H A DMultiBuffer.cpp123 AffineExpr induc = getAffineDimExpr(0, allocOp.getContext()); in multiBuffer()
124 AffineExpr init = getAffineDimExpr(1, allocOp.getContext()); in multiBuffer()
125 AffineExpr step = getAffineDimExpr(2, allocOp.getContext()); in multiBuffer()
H A DFoldSubViewOps.cpp67 AffineExpr expr = rewriter.getAffineDimExpr(0); in resolveSourceIndices()
119 exprs.push_back(getAffineDimExpr(dim, context)); in getPermutationMapAttr()
/llvm-project-15.0.7/mlir/lib/Dialect/SCF/Transforms/
H A DParallelLoopTiling.cpp83 {getAffineDimExpr(/*position=*/0, b.getContext()), in tileParallelLoop()
84 getAffineDimExpr(/*position=*/1, b.getContext()) - in tileParallelLoop()
85 getAffineDimExpr(/*position=*/2, b.getContext())}, in tileParallelLoop()
/llvm-project-15.0.7/mlir/lib/Dialect/SCF/Utils/
H A DAffineCanonicalizationUtils.cpp241 ivUb = rewriter.getAffineDimExpr(dimLb) + 1; in addLoopRangeConstraints()
246 : rewriter.getAffineDimExpr(dimLb); in addLoopRangeConstraints()
248 : rewriter.getAffineDimExpr(dimUb); in addLoopRangeConstraints()
/llvm-project-15.0.7/mlir/lib/IR/
H A DBuilders.cpp285 AffineExpr Builder::getAffineDimExpr(unsigned position) { in getAffineDimExpr() function in Builder
286 return mlir::getAffineDimExpr(position, context); in getAffineDimExpr()
305 return AffineMap::get(/*dimCount=*/1, /*symbolCount=*/0, getAffineDimExpr(0)); in getDimIdentityMap()
312 dimExprs.push_back(getAffineDimExpr(i)); in getMultiDimIdentityMap()
324 auto expr = getAffineDimExpr(0) + shift; in getSingleDimShiftAffineMap()
H A DAffineMap.cpp211 affExprs.push_back(getAffineDimExpr(index, context)); in getPermutationMap()
249 dimExprs.push_back(mlir::getAffineDimExpr(i, context)); in getMultiDimIdentityMap()
447 newDims[idx] = getAffineDimExpr(idx, getContext()); in compose()
553 dimReplacements.push_back(getAffineDimExpr(numDims++, context)); in compressDims()
664 exprs[d.getPosition()] = getAffineDimExpr(en.index(), d.getContext()); in inversePermutation()
693 exprs[map.getDimPosition(i)] = getAffineDimExpr(i, context); in inverseAndBroadcastProjectedPermutation()
H A DAffineExpr.cpp112 dims.push_back(getAffineDimExpr(idx, getContext())); in shiftDims()
114 dims.push_back(getAffineDimExpr(idx + shift, getContext())); in shiftDims()
282 return *this == mlir::getAffineDimExpr(position, getContext()); in isFunctionOfDim()
489 AffineExpr mlir::getAffineDimExpr(unsigned position, MLIRContext *context) { in getAffineDimExpr() function in mlir
905 auto id = j < numDims ? getAffineDimExpr(j, context) in getAffineExprFromFlatForm()
998 addEntry(indexEntry, flatExprs[j], getAffineDimExpr(j, context)); in getSemiAffineExprFromFlatForm()
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Utils/
H A DLoopUtils.cpp92 newUbExprs[i] = b.getAffineDimExpr(0) + b.getAffineDimExpr(i + 1); in getCleanupLoopLowerBound()
835 AffineExpr dim = b.getAffineDimExpr(origUbMap.getNumDims()); in constructTiledIndexSetHyperRect()
847 AffineExpr dim = b.getAffineDimExpr(0); in constructTiledIndexSetHyperRect()
1137 auto d0 = b.getAffineDimExpr(0); in loopUnrollByFactor()
1327 auto d0 = builder.getAffineDimExpr(0); in loopUnrollJamByFactor()
1764 builder.getAffineDimExpr(0) * in coalesceLoops()
1794 builder.getAffineDimExpr(0).floorDiv( in coalesceLoops()
1983 fastBufExprs.push_back(b.getAffineDimExpr(2 * d + 1) - in generatePointWiseCopy()
1984 b.getAffineDimExpr(2 * d)); in generatePointWiseCopy()
2127 offset = offset + lbs[d][j] * top.getAffineDimExpr(j); in generateCopy()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorDropLeadUnitDim.cpp349 auto targetExpr = rewriter.getAffineDimExpr(currDim); in matchAndRewrite()
353 auto targetExpr = rewriter.getAffineDimExpr(currDim); in matchAndRewrite()
378 auto targetExpr = rewriter.getAffineDimExpr( in matchAndRewrite()
H A DVectorTransferOpTransforms.cpp453 getAffineDimExpr(firstContiguousInnerDim, rewriter.getContext())}; in matchAndRewrite()
511 getAffineDimExpr(firstContiguousInnerDim, rewriter.getContext())}; in matchAndRewrite()
/llvm-project-15.0.7/mlir/include/mlir/IR/
H A DAffineExpr.h251 AffineExpr getAffineDimExpr(unsigned position, MLIRContext *context);
311 e = getAffineDimExpr(N, ctx); in bindDims()
/llvm-project-15.0.7/mlir/lib/Conversion/TosaToLinalg/
H A DTosaToLinalg.cpp637 mlir::getAffineDimExpr(it.index(), rewriter.getContext())); in elementwiseMatchAndRewriteHelper()
830 srcExprs.push_back(mlir::getAffineDimExpr(i, rewriter.getContext())); in reduceMatchAndRewriteHelper()
921 exprs.push_back(rewriter.getAffineDimExpr(i)); in createReassociationMapsForCollapse()
938 rewriter.getAffineDimExpr(currSrcDim++)); in createReassociationMapsForCollapse()
943 rewriter.getAffineDimExpr(currSrcDim++)); in createReassociationMapsForCollapse()
949 rewriter.getAffineDimExpr(currSrcDim++)); in createReassociationMapsForCollapse()
1125 inputExprs[value] = rewriter.getAffineDimExpr(index); in matchAndRewrite()
1205 rewriter.getAffineDimExpr(rank - 1)}; in matchAndRewrite()
1228 rewriter.getAffineDimExpr(rank - 1)}; in matchAndRewrite()
1809 dimExprs.push_back(rewriter.getAffineDimExpr(i * 2 + 1)); in matchAndRewrite()
[all …]
H A DTosaToLinalgNamed.cpp154 reassociationMap[i].push_back(rewriter.getAffineDimExpr(i)); in createDepthwiseConvCollapseMap()
157 rewriter.getAffineDimExpr(outputRank)); in createDepthwiseConvCollapseMap()
267 {rewriter.getAffineDimExpr(3)}, rewriter.getContext())); in matchAndRewrite()
424 {rewriter.getAffineDimExpr(3)}, rewriter.getContext())); in matchAndRewrite()
594 {rewriter.getAffineDimExpr(1)}, in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/CAPI/IR/
H A DAffineExpr.cpp73 return wrap(getAffineDimExpr(position, unwrap(ctx))); in mlirAffineDimExprGet()
/llvm-project-15.0.7/flang/lib/Optimizer/Transforms/
H A DAffinePromotion.cpp227 return {mlir::getAffineDimExpr(dimCount++, value.getContext())}; in toAffineExpr()
316 mlir::AffineExpr idx = mlir::getAffineDimExpr(i, context), in createArrayIndexAffineMap()
516 (lowerBound + mlir::getAffineDimExpr(0, op.getContext())) * in genericBounds()
/llvm-project-15.0.7/mlir/lib/Conversion/LinalgToSPIRV/
H A DLinalgToSPIRV.cpp99 if (inputMap != AffineMap::get(1, 0, getAffineDimExpr(0, op->getContext()))) in matchAsPerformingReduction()
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Analysis/
H A DAffineStructures.cpp216 getAffineDimExpr(lbsStart + ivIdx, ctx)); in getHyperrectangular()
221 getAffineDimExpr(ubsStart + ivIdx, ctx)); in getHyperrectangular()
993 memo[i] = getAffineDimExpr(i, context); in getSliceBounds()
995 memo[i] = getAffineDimExpr(i - num, context); in getSliceBounds()
1452 memo[i] = getAffineDimExpr(i, context); in computeLocalVars()
1595 builder.getAffineDimExpr(std::distance(dims.begin(), dimIt)); in alignAffineMapWithValues()
/llvm-project-15.0.7/mlir/lib/AsmParser/
H A DAffineParser.cpp313 : getAffineDimExpr(numDimOperands++, getContext()); in parseSSAIdExpr()
490 auto dimension = getAffineDimExpr(numDims++, getContext()); in parseDimIdList()
/llvm-project-15.0.7/mlir/lib/Conversion/SCFToGPU/
H A DSCFToGPU.cpp450 rewriter.getAffineDimExpr(0) * rewriter.getAffineSymbolExpr(0) + in processParallelLoop()
494 ((rewriter.getAffineDimExpr(0) - rewriter.getAffineSymbolExpr(0)) in processParallelLoop()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Utils/
H A DVectorUtils.cpp162 perm[kvp.second] = getAffineDimExpr(dim, context); in makePermutationMap()
/llvm-project-15.0.7/mlir/lib/Dialect/Utils/
H A DReshapeOpsUtils.cpp139 reassociationMap.push_back(mlir::getAffineDimExpr(index, context)); in convertReassociationIndicesToExprs()
/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/
H A DTensorInferTypeOpInterfaceImpl.cpp182 AffineExpr expr = b.getAffineDimExpr(0); in reifyResultShapes()

12