Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dloopnest.cpp1336 auto for_to_split = fors.back(); in optimizeConditionals() local
1337 if (!LoopNest::isNormalized(for_to_split)) { in optimizeConditionals()
1342 if (split_fors.count(for_to_split)) { in optimizeConditionals()
1354 split_fors.insert(for_to_split); in optimizeConditionals()
1358 comp_values.push_back(for_to_split->stop()); in optimizeConditionals()
1373 for_to_split->body()->clone_and_replace(store, new_store); in optimizeConditionals()
1375 for_to_split->var(), in optimizeConditionals()
1382 auto par = to<Block>(for_to_split->get_parent()); in optimizeConditionals()
1383 par->replace_stmt(for_to_split, alloc<Block>(split_loops)); in optimizeConditionals()