118           SmallVector<AffineForOp, 4> loops;  in getLastDependentOpInRange()
121           if (llvm::is_contained(loops, cast<AffineForOp>(opB))) {  in getLastDependentOpInRange()
138 static Operation *getFusedLoopNestInsertionPoint(AffineForOp srcForOp,  in getFusedLoopNestInsertionPoint()
139                                                  AffineForOp dstForOp) {  in getFusedLoopNestInsertionPoint()
179 gatherLoadsAndStores(AffineForOp forOp,  in gatherLoadsAndStores()
260 FusionResult mlir::canFuseLoops(AffineForOp srcForOp, AffineForOp dstForOp,  in canFuseLoops()
369 LogicalResult promoteSingleIterReductionLoop(AffineForOp forOp,  in promoteSingleIterReductionLoop()
377   if (!isa<AffineForOp>(parentOp))  in promoteSingleIterReductionLoop()
382   AffineForOp parentForOp = forOp->getParentOfType<AffineForOp>();  in promoteSingleIterReductionLoop()
383   AffineForOp newLoop = replaceForOpWithNewYields(  in promoteSingleIterReductionLoop()
432 void mlir::fuseLoops(AffineForOp srcForOp, AffineForOp dstForOp,  in fuseLoops()
441   SmallVector<AffineForOp, 4> sliceLoops;  in fuseLoops()
466   for (AffineForOp forOp : sliceLoops) {  in fuseLoops()
481 bool mlir::getLoopNestStats(AffineForOp forOpRoot, LoopNestStats *stats) {  in getLoopNestStats()
482   auto walkResult = forOpRoot.walk([&](AffineForOp forOp) {  in getLoopNestStats()
486       if (!isa<AffineForOp>(parentForOp)) {  in getLoopNestStats()
498       if (!isa<AffineForOp, AffineIfOp>(op))  in getLoopNestStats()
569 int64_t mlir::getComputeCost(AffineForOp forOp, LoopNestStats &stats) {  in getComputeCost()
580 bool mlir::getFusionComputeCost(AffineForOp srcForOp, LoopNestStats &srcStats,  in getFusionComputeCost()
581                                 AffineForOp dstForOp, LoopNestStats &dstStats,  in getFusionComputeCost()
617           SmallVector<AffineForOp, 4> loops;  in getFusionComputeCost()
621           if (llvm::is_contained(loops, cast<AffineForOp>(insertPointParent))) {  in getFusionComputeCost()
623                     dyn_cast_or_null<AffineForOp>(user->getParentOp())) {  in getFusionComputeCost()