35 void mlir::getLoopIVs(Operation &op, SmallVectorImpl<AffineForOp> *loops) {  in getLoopIVs()
37   AffineForOp currAffineForOp;  in getLoopIVs()
41     if (AffineForOp currAffineForOp = dyn_cast<AffineForOp>(currOp))  in getLoopIVs()
59     if (isa<AffineIfOp, AffineForOp>(currOp))  in getEnclosingAffineForAndIfOps()
73     AffineForOp loop = getForInductionVarOwner(iv);  in getSourceAsConstraints()
179     AffineForOp dstLoop =  in isSliceMaximalFastCheck()
187     AffineForOp srcLoop = getForInductionVarOwner(ivs[i]);  in isSliceMaximalFastCheck()
303     AffineForOp loop = getForInductionVarOwner(iv);  in isMaximal()
464     SmallVector<AffineForOp, 4> ivs;  in compute()
558   SmallVector<AffineForOp, 4> enclosingIVs;  in compute()
565     AffineForOp iv;  in compute()
748     if (auto childAffineForOp = dyn_cast<AffineForOp>(op))  in getInstAtPosition()
780     ArrayRef<Operation *> ops, SmallVectorImpl<AffineForOp> *surroundingLoops) {  in getInnermostCommonLoopDepth()
784   std::vector<SmallVector<AffineForOp, 4>> loops(numOps);  in getInnermostCommonLoopDepth()
917   SmallVector<AffineForOp, 4> surroundingLoops;  in computeSliceUnion()
999     AffineForOp forOp = getForInductionVarOwner(slice.ivs[i]);  in buildSliceTripCountMap()
1050   SmallVector<AffineForOp, 4> srcLoopIVs;  in getComputationSliceState()
1055   SmallVector<AffineForOp, 4> dstLoopIVs;  in getComputationSliceState()
1159 AffineForOp
1164   SmallVector<AffineForOp, 4> srcLoopIVs;  in insertBackwardComputationSlice()
1169   SmallVector<AffineForOp, 4> dstLoopIVs;  in insertBackwardComputationSlice()
1174     return AffineForOp();  in insertBackwardComputationSlice()
1187       cast<AffineForOp>(b.clone(*srcLoopIVs[0].getOperation()));  in insertBackwardComputationSlice()
1192   SmallVector<AffineForOp, 4> sliceSurroundingLoops;  in insertBackwardComputationSlice()
1245     if (isa<AffineForOp>(currOp))  in getNestingDepth()
1272   SmallVector<AffineForOp, 4> loopsA, loopsB;  in getNumCommonSurroundingLoops()
1330 Optional<int64_t> mlir::getMemoryFootprintBytes(AffineForOp forOp,  in getMemoryFootprintBytes()
1339 bool mlir::isLoopParallelAndContainsReduction(AffineForOp forOp) {  in isLoopParallelAndContainsReduction()
1348 void mlir::getSequentialLoops(AffineForOp forOp,  in getSequentialLoops()
1351     if (auto innerFor = dyn_cast<AffineForOp>(op))  in getSequentialLoops()