Searched refs:normalized_loops (Results 1 – 1 of 1) sorted by relevance
2434 auto normalized_loops = getLoopStmtsInLoopNest(loops.front(), loops.size()); in flatten() local2437 normalized_loops[0]->var()->name_hint() + "_flat", in flatten()2438 normalized_loops[0]->var()->dtype()); in flatten()2441 for (size_t i = 0; i < normalized_loops.size(); ++i) { in flatten()2442 size_t idx = normalized_loops.size() - i - 1; in flatten()2443 auto curr_loop = normalized_loops[idx]; in flatten()2450 Substitute(normalized_loops.back()->removeBody(), var_mapping); in flatten()2452 normalized_loops.front()->set_var(flat_var); in flatten()2453 normalized_loops.front()->set_start(immLike(stop, 0)); in flatten()2454 normalized_loops.front()->set_stop(stop); in flatten()[all …]