Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dloopnest_randomization.cpp39 std::vector<std::vector<ForPtr>> all_nested_loops; in GetAllPerfectlyNestedLoopNests() local
42 return all_nested_loops; in GetAllPerfectlyNestedLoopNests()
53 all_nested_loops.push_back(nested_loops); in GetAllPerfectlyNestedLoopNests()
59 return all_nested_loops; in GetAllPerfectlyNestedLoopNests()
435 auto all_nested_loops = in loopnestRandomization() local
437 if (all_nested_loops.empty()) { in loopnestRandomization()
442 int index = rand() % (int)all_nested_loops.size(); in loopnestRandomization()
443 auto nested_loops = all_nested_loops.at(index); in loopnestRandomization()
476 auto all_nested_loops = in loopnestRandomization() local
478 if (all_nested_loops.empty()) { in loopnestRandomization()
[all …]