Lines Matching refs:depthToLoops
2596 std::vector<SmallVector<AffineForOp, 2>> &depthToLoops) { in gatherLoopsInBlock() argument
2598 assert(currLoopDepth <= depthToLoops.size() && "Unexpected currLoopDepth"); in gatherLoopsInBlock()
2599 if (currLoopDepth == depthToLoops.size()) in gatherLoopsInBlock()
2600 depthToLoops.emplace_back(); in gatherLoopsInBlock()
2604 depthToLoops[currLoopDepth].push_back(forOp); in gatherLoopsInBlock()
2605 gatherLoopsInBlock(forOp.getBody(), currLoopDepth + 1, depthToLoops); in gatherLoopsInBlock()
2612 std::vector<SmallVector<AffineForOp, 2>> &depthToLoops) { in gatherLoops() argument
2614 gatherLoopsInBlock(&block, /*currLoopDepth=*/0, depthToLoops); in gatherLoops()
2617 if (!depthToLoops.empty()) { in gatherLoops()
2618 assert(depthToLoops.back().empty() && "Last loop level is not empty?"); in gatherLoops()
2619 depthToLoops.pop_back(); in gatherLoops()