117           SmallVector<AffineForOp, 4> loops;  in getLastDependentOpInRange()
120           if (llvm::is_contained(loops, cast<AffineForOp>(opB))) {  in getLastDependentOpInRange()
137 static Operation *getFusedLoopNestInsertionPoint(AffineForOp srcForOp,  in getFusedLoopNestInsertionPoint()
138                                                  AffineForOp dstForOp) {  in getFusedLoopNestInsertionPoint()
178 gatherLoadsAndStores(AffineForOp forOp,  in gatherLoadsAndStores()
263 FusionResult mlir::canFuseLoops(AffineForOp srcForOp, AffineForOp dstForOp,  in canFuseLoops()
367 void mlir::fuseLoops(AffineForOp srcForOp, AffineForOp dstForOp,  in fuseLoops()
375   SmallVector<AffineForOp, 4> sliceLoops;  in fuseLoops()
395   for (AffineForOp forOp : sliceLoops)  in fuseLoops()
402 bool mlir::getLoopNestStats(AffineForOp forOpRoot, LoopNestStats *stats) {  in getLoopNestStats()
403   auto walkResult = forOpRoot.walk([&](AffineForOp forOp) {  in getLoopNestStats()
407       if (!isa<AffineForOp>(parentForOp)) {  in getLoopNestStats()
419       if (!isa<AffineForOp, AffineIfOp>(op))  in getLoopNestStats()
523     AffineForOp forOp = getForInductionVarOwner(slice.ivs[i]);  in buildSliceTripCountMap()
554 int64_t mlir::getComputeCost(AffineForOp forOp, LoopNestStats &stats) {  in getComputeCost()
565 bool mlir::getFusionComputeCost(AffineForOp srcForOp, LoopNestStats &srcStats,  in getFusionComputeCost()
566                                 AffineForOp dstForOp, LoopNestStats &dstStats,  in getFusionComputeCost()
602           SmallVector<AffineForOp, 4> loops;  in getFusionComputeCost()
606           if (llvm::is_contained(loops, cast<AffineForOp>(insertPointParent))) {  in getFusionComputeCost()
608                     dyn_cast_or_null<AffineForOp>(user->getParentOp())) {  in getFusionComputeCost()