Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dloopnest.h294 static std::vector<ForPtr> distributeLoopOverInnerLoops(const ForPtr& loop);
H A Dloopnest_randomization.cpp345 l.distributeLoopOverInnerLoops(loop); in loopnestRandomization()
H A Dtensorexpr_init.cpp592 return LoopNest::distributeLoopOverInnerLoops(f); in initTensorExprBindings()
H A Dloopnest.cpp1742 std::vector<ForPtr> LoopNest::distributeLoopOverInnerLoops(const ForPtr& loop) { in distributeLoopOverInnerLoops() function in torch::jit::tensorexpr::LoopNest
1751 auto result = distributeLoopOverInnerLoops(loop); in distributeLoopAndParentsOverInnerLoops()
/aosp_15_r20/external/pytorch/test/cpp/tensorexpr/
H A Dtest_loopnest.cpp5206 auto new_loops = LoopNest::distributeLoopOverInnerLoops(forI); in TEST()