Home
last modified time | relevance | path

Searched refs:normalized_loops (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dloopnest.cpp2434 auto normalized_loops = getLoopStmtsInLoopNest(loops.front(), loops.size()); in flatten() local
2437 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 …]