33 void mlir::getLoopIVs(Operation &op, SmallVectorImpl<AffineForOp> *loops) {  in getLoopIVs()
35   AffineForOp currAffineForOp;  in getLoopIVs()
38   while (currOp && ((currAffineForOp = dyn_cast<AffineForOp>(currOp)) ||  in getLoopIVs()
57   while (currOp && (isa<AffineIfOp, AffineForOp>(currOp))) {  in getEnclosingAffineForAndIfOps()
252     SmallVector<AffineForOp, 4> ivs;  in compute()
346   SmallVector<AffineForOp, 4> enclosingIVs;  in compute()
353     AffineForOp iv;  in compute()
537     if (auto childAffineForOp = dyn_cast<AffineForOp>(op))  in getInstAtPosition()
569     ArrayRef<Operation *> ops, SmallVectorImpl<AffineForOp> *surroundingLoops) {  in getInnermostCommonLoopDepth()
573   std::vector<SmallVector<AffineForOp, 4>> loops(numOps);  in getInnermostCommonLoopDepth()
707   SmallVector<AffineForOp, 4> surroundingLoops;  in computeSliceUnion()
762   SmallVector<AffineForOp, 4> srcLoopIVs;  in getComputationSliceState()
767   SmallVector<AffineForOp, 4> dstLoopIVs;  in getComputationSliceState()
854 AffineForOp
859   SmallVector<AffineForOp, 4> srcLoopIVs;  in insertBackwardComputationSlice()
864   SmallVector<AffineForOp, 4> dstLoopIVs;  in insertBackwardComputationSlice()
869     return AffineForOp();  in insertBackwardComputationSlice()
882       cast<AffineForOp>(b.clone(*srcLoopIVs[0].getOperation()));  in insertBackwardComputationSlice()
887   SmallVector<AffineForOp, 4> sliceSurroundingLoops;  in insertBackwardComputationSlice()
948     if (isa<AffineForOp>(currOp))  in getNestingDepth()
975   SmallVector<AffineForOp, 4> loopsA, loopsB;  in getNumCommonSurroundingLoops()
1033 Optional<int64_t> mlir::getMemoryFootprintBytes(AffineForOp forOp,  in getMemoryFootprintBytes()
1043 void mlir::getSequentialLoops(AffineForOp forOp,  in getSequentialLoops()
1046     if (auto innerFor = dyn_cast<AffineForOp>(op))  in getSequentialLoops()
1053 bool mlir::isLoopParallel(AffineForOp forOp) {  in isLoopParallel()
1059     else if (!isa<AffineForOp, AffineYieldOp, AffineIfOp>(opInst) &&  in isLoopParallel()