Home
last modified time | relevance | path

Searched defs:forOp (Results 1 – 25 of 28) sorted by relevance

12

/external/llvm-project/mlir/lib/Dialect/Affine/Transforms/
DLoopUnroll.cpp65 static bool isInnermostAffineForOp(AffineForOp forOp) { in isInnermostAffineForOp()
79 f.walk([&](AffineForOp forOp) { in gatherInnermostLoops()
93 getFunction().walk([&](AffineForOp forOp) { in runOnFunction()
98 for (auto forOp : loops) in runOnFunction() local
112 for (auto forOp : loops) in runOnFunction() local
122 LogicalResult LoopUnroll::runOnAffineForOp(AffineForOp forOp) { in runOnAffineForOp()
DAffineDataCopyGeneration.cpp130 AffineForOp forOp; in runOnBlock() local
138 auto exceedsCapacity = [&](AffineForOp forOp) { in runOnBlock()
219 if (auto forOp = dyn_cast<AffineForOp>(op)) in runOnFunction() local
DLoopUnrollAndJam.cpp76 if (auto forOp = dyn_cast<AffineForOp>(entryBlock.front())) in runOnFunction() local
DAffineLoopInvariantCodeMotion.cpp191 void LoopInvariantCodeMotion::runOnAffineForOp(AffineForOp forOp) { in runOnAffineForOp()
DSuperVectorize.cpp834 return [&parallelLoops, fastestVaryingMemRefDimension](Operation &forOp) { in isVectorizableLoopPtrFactory()
/external/llvm-project/mlir/lib/Conversion/SCFToGPU/
DSCFToGPU.cpp55 static Operation::operand_range getLowerBoundOperands(AffineForOp forOp) { in getLowerBoundOperands()
60 static Operation::operand_range getUpperBoundOperands(AffineForOp forOp) { in getUpperBoundOperands()
66 static Value getOrCreateStep(AffineForOp forOp, OpBuilder &builder) { in getOrCreateStep()
72 static Value getOrEmitLowerBound(AffineForOp forOp, OpBuilder &builder) { in getOrEmitLowerBound()
78 static Value getOrEmitUpperBound(AffineForOp forOp, OpBuilder &builder) { in getOrEmitUpperBound()
88 static LogicalResult checkAffineLoopNestMappableImpl(AffineForOp forOp, in checkAffineLoopNestMappableImpl()
114 static LogicalResult checkAffineLoopNestMappable(AffineForOp forOp, in checkAffineLoopNestMappable()
164 AffineLoopToGpuConverter::collectBounds(AffineForOp forOp, unsigned numLoops) { in collectBounds()
263 static LogicalResult convertAffineLoopNestToGPULaunch(AffineForOp forOp, in convertAffineLoopNestToGPULaunch()
279 LogicalResult mlir::convertAffineLoopNestToGPULaunch(AffineForOp forOp, in convertAffineLoopNestToGPULaunch()
DSCFToGPUPass.cpp37 if (auto forOp = dyn_cast<AffineForOp>(&op)) { in runOnFunction() local
/external/llvm-project/mlir/lib/Transforms/
DPipelineDataTransfer.cpp64 static bool doubleBuffer(Value oldMemRef, AffineForOp forOp) { in doubleBuffer()
135 getFunction().walk([&](AffineForOp forOp) { forOps.push_back(forOp); }); in runOnFunction()
136 for (auto forOp : forOps) in runOnFunction() local
165 AffineForOp forOp, in findMatchingStartFinishInsts()
236 void PipelineDataTransfer::runOnAffineForOp(AffineForOp forOp) { in runOnAffineForOp()
DLoopFusion.cpp642 if (auto forOp = dyn_cast<AffineForOp>(op)) { in init() local
775 static Value createPrivateMemRef(AffineForOp forOp, Operation *srcStoreOpInst, in createPrivateMemRef()
/external/llvm-project/mlir/lib/Transforms/Utils/
DLoopFusionUtils.cpp178 gatherLoadsAndStores(AffineForOp forOp, in gatherLoadsAndStores()
380 auto forOp = getForInductionVarOwner(loopIV); in fuseLoops() local
395 for (AffineForOp forOp : sliceLoops) in fuseLoops() local
403 auto walkResult = forOpRoot.walk([&](AffineForOp forOp) { in getLoopNestStats()
455 Operation *forOp, LoopNestStats &stats, in getComputeCostHelper()
523 AffineForOp forOp = getForInductionVarOwner(slice.ivs[i]); in buildSliceTripCountMap() local
554 int64_t mlir::getComputeCost(AffineForOp forOp, LoopNestStats &stats) { in getComputeCost()
607 if (auto forOp = in getFusionComputeCost() local
DLoopUtils.cpp58 static void getCleanupLoopLowerBound(AffineForOp forOp, unsigned unrollFactor, in getCleanupLoopLowerBound()
153 LogicalResult mlir::promoteIfSingleIteration(AffineForOp forOp) { in promoteIfSingleIteration()
195 LogicalResult mlir::promoteIfSingleIteration(scf::ForOp forOp) { in promoteIfSingleIteration()
236 if (auto forOp = dyn_cast<AffineForOp>(op)) in promoteSingleIterationLoops() local
238 else if (auto forOp = dyn_cast<scf::ForOp>(op)) in promoteSingleIterationLoops() local
305 LogicalResult mlir::affineForOpBodySkew(AffineForOp forOp, in affineForOpBodySkew()
1031 LogicalResult mlir::loopUnrollFull(AffineForOp forOp) { in loopUnrollFull()
1044 LogicalResult mlir::loopUnrollUpToFactor(AffineForOp forOp, in loopUnrollUpToFactor()
1099 LogicalResult mlir::loopUnrollByFactor(AffineForOp forOp, in loopUnrollByFactor()
1163 LogicalResult mlir::loopUnrollByFactor(scf::ForOp forOp, in loopUnrollByFactor()
[all …]
/external/llvm-project/mlir/lib/Analysis/
DLoopAnalysis.cpp34 AffineForOp forOp, AffineMap *tripCountMap, in buildTripCountMapAndOperands()
87 Optional<uint64_t> mlir::getConstantTripCount(AffineForOp forOp) { in getConstantTripCount()
113 uint64_t mlir::getLargestDivisorOfTripCount(AffineForOp forOp) { in getLargestDivisorOfTripCount()
281 auto *forOp = loop.getOperation(); in isVectorizableLoopBodyWithOpCond() local
349 bool mlir::isOpwiseShiftValid(AffineForOp forOp, ArrayRef<uint64_t> shifts) { in isOpwiseShiftValid()
DSliceAnalysis.cpp44 if (auto forOp = dyn_cast<AffineForOp>(op)) { in getForwardSliceImpl() local
48 } else if (auto forOp = dyn_cast<scf::ForOp>(op)) { in getForwardSliceImpl() local
DUtils.cpp900 auto forOp = sliceSurroundingLoops[dstLoopDepth + i]; in insertBackwardComputationSlice() local
1033 Optional<int64_t> mlir::getMemoryFootprintBytes(AffineForOp forOp, in getMemoryFootprintBytes()
1043 void mlir::getSequentialLoops(AffineForOp forOp, in getSequentialLoops()
1053 bool mlir::isLoopParallel(AffineForOp forOp) { in isLoopParallel()
DAffineAnalysis.cpp96 if (AffineForOp forOp = dyn_cast<AffineForOp>(op)) in getIndexSet() local
104 if (AffineForOp forOp = dyn_cast<AffineForOp>(op)) { in getIndexSet() local
631 AffineForOp forOp = getForInductionVarOwner(commonForIV); in getCommonBlock() local
953 AffineForOp forOp, unsigned maxLoopDepth, in getDependenceComponents()
/external/llvm-project/mlir/lib/Dialect/Affine/Utils/
DUtils.cpp50 if (auto forOp = dyn_cast<AffineForOp>(parentOp)) { in getOutermostInvariantForOp() local
134 void mlir::affineParallelize(AffineForOp forOp) { in affineParallelize()
/external/llvm-project/mlir/test/lib/Dialect/Affine/
DTestLoopPermutation.cpp49 if (auto forOp = dyn_cast<AffineForOp>(op)) { in runOnFunction() local
DTestAffineDataCopy.cpp98 if (auto forOp = dyn_cast<AffineForOp>(op)) in runOnFunction() local
/external/llvm-project/mlir/test/lib/Transforms/
DTestLoopUnrolling.cpp47 func.walk([&](scf::ForOp forOp) { in runOnFunction()
DTestVectorTransforms.cpp265 auto forOp = builder.create<scf::ForOp>(op.getLoc(), zero, numIter, one); in runOnFunction() local
/external/llvm-project/mlir/lib/Conversion/SCFToStandard/
DSCFToStandard.cpp282 LogicalResult ForLowering::matchAndRewrite(ForOp forOp, in matchAndRewrite()
423 ForOp forOp = rewriter.create<ForOp>(loc, lower, upper, step, iterArgs); in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dmhlo_control_flow_to_scf.cc131 auto forOp = b.create<mlir::scf::ForOp>(whileOp.getLoc(), forloopIndVar, in MatchAndRewrite() local
/external/llvm-project/mlir/lib/Conversion/SCFToSPIRV/
DSCFToSPIRV.cpp123 ForOpConversion::matchAndRewrite(scf::ForOp forOp, ArrayRef<Value> operands, in matchAndRewrite()
/external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
DTransforms.cpp418 if (auto forOp = scf::getForInductionVarOwner(dim)) in substitute() local
DFusion.cpp481 if (auto forOp = blockArg.getDefiningOp<scf::ForOp>()) { in getProducerOfTensor() local

12