79   SmallVector<AffineForOp, 4> forOps;
86       if (isa<AffineForOp>(op))  in collect()
87         forOps.push_back(cast<AffineForOp>(op));  in collect()
216   Node *getForOpNode(AffineForOp forOp) {  in getForOpNode()
567       if (!isa<AffineForOp>(getNode(edge.id)->op))  in forEachMemRefEdge()
677     if (!isa<AffineForOp>(srcNode->op))  in getProducerCandidates()
745     if (auto forOp = dyn_cast<AffineForOp>(op)) {  in init()
829         SmallVector<AffineForOp, 4> loops;  in init()
865   assert(isa<AffineForOp>(node->op));  in sinkSequentialLoops()
866   AffineForOp newRootForOp = sinkSequentialLoops(cast<AffineForOp>(node->op));  in sinkSequentialLoops()
890 static Value createPrivateMemRef(AffineForOp forOp, Operation *srcStoreOpInst,  in createPrivateMemRef()
1034     if (isa<AffineForOp>(op))  in hasNonAffineUsersOnThePath()
1089                                AffineForOp dstForOp,  in isFusionProfitable()
1106   SmallVector<AffineForOp, 4> srcLoopIVs;  in isFusionProfitable()
1432       if (!isa<AffineForOp>(dstNode->op))  in fuseProducerConsumerNodes()
1446       auto dstAffineForOp = cast<AffineForOp>(dstNode->op);  in fuseProducerConsumerNodes()
1464           auto srcAffineForOp = cast<AffineForOp>(srcNode->op);  in fuseProducerConsumerNodes()
1470           if (isa<AffineForOp>(srcNode->op) && srcNode->op->getNumResults() > 0)  in fuseProducerConsumerNodes()
1708       if (!isa<AffineForOp>(dstNode->op))  in fuseSiblingNodes()
1719     auto dstAffineForOp = cast<AffineForOp>(dstNode->op);  in fuseWithSiblingNodes()
1753       SmallVector<AffineForOp, 4> dstLoopIVs;  in fuseWithSiblingNodes()
1756       auto sibAffineForOp = cast<AffineForOp>(sibNode->op);  in fuseWithSiblingNodes()
1797       auto dstForInst = cast<AffineForOp>(dstNode->op);  in fuseWithSiblingNodes()
1858           SmallVector<AffineForOp, 4> loops;  in findSiblingNodeToFuse()
1911             if (!isa<AffineForOp>(sibNode->op))  in findSiblingNodeToFuse()
1938     auto dstForInst = cast<AffineForOp>(dstNode->op);  in updateStateAfterSiblingFusion()