Lines Matching defs:depthToLoops
2499 /// Gathers all AffineForOps in 'block' at 'currLoopDepth' in 'depthToLoops'.
2502 std::vector<SmallVector<AffineForOp, 2>> &depthToLoops) {
2504 assert(currLoopDepth <= depthToLoops.size() && "Unexpected currLoopDepth");
2505 if (currLoopDepth == depthToLoops.size())
2506 depthToLoops.emplace_back();
2510 depthToLoops[currLoopDepth].push_back(forOp);
2511 gatherLoopsInBlock(forOp.getBody(), currLoopDepth + 1, depthToLoops);
2518 func::FuncOp func, std::vector<SmallVector<AffineForOp, 2>> &depthToLoops) {
2520 gatherLoopsInBlock(&block, /*currLoopDepth=*/0, depthToLoops);
2523 if (!depthToLoops.empty()) {
2524 assert(depthToLoops.back().empty() && "Last loop level is not empty?");
2525 depthToLoops.pop_back();