82   SmallVector<AffineForOp, 4> forOps;
89       if (isa<AffineForOp>(op))  in collect()
90         forOps.push_back(cast<AffineForOp>(op));  in collect()
219   Node *getForOpNode(AffineForOp forOp) {  in getForOpNode()
600       if (!isa<AffineForOp>(getNode(edge.id)->op))  in forEachMemRefEdge()
642     if (auto forOp = dyn_cast<AffineForOp>(op)) {  in init()
698         SmallVector<AffineForOp, 4> loops;  in init()
750   assert(isa<AffineForOp>(node->op));  in sinkSequentialLoops()
751   AffineForOp newRootForOp = sinkSequentialLoops(cast<AffineForOp>(node->op));  in sinkSequentialLoops()
775 static Value createPrivateMemRef(AffineForOp forOp, Operation *srcStoreOpInst,  in createPrivateMemRef()
920     if (isa<AffineForOp>(op))  in hasNonAffineUsersOnThePath()
1064   SmallVector<AffineForOp, 4> srcLoopIVs;  in isFusionProfitable()
1073   SmallVector<AffineForOp, 4> dstLoopIVs;  in isFusionProfitable()
1387       if (!isa<AffineForOp>(dstNode->op))  in fuseProducerConsumerNodes()
1425           if (!isa<AffineForOp>(srcNode->op))  in fuseProducerConsumerNodes()
1487           auto srcAffineForOp = cast<AffineForOp>(srcNode->op);  in fuseProducerConsumerNodes()
1488           auto dstAffineForOp = cast<AffineForOp>(dstNode->op);  in fuseProducerConsumerNodes()
1647       if (!isa<AffineForOp>(dstNode->op))  in fuseSiblingNodes()
1658     auto dstAffineForOp = cast<AffineForOp>(dstNode->op);  in fuseWithSiblingNodes()
1692       SmallVector<AffineForOp, 4> dstLoopIVs;  in fuseWithSiblingNodes()
1695       auto sibAffineForOp = cast<AffineForOp>(sibNode->op);  in fuseWithSiblingNodes()
1732       auto dstForInst = cast<AffineForOp>(dstNode->op);  in fuseWithSiblingNodes()
1793           SmallVector<AffineForOp, 4> loops;  in findSiblingNodeToFuse()
1846             if (!isa<AffineForOp>(sibNode->op))  in findSiblingNodeToFuse()
1873     auto dstForInst = cast<AffineForOp>(dstNode->op);  in updateStateAfterSiblingFusion()