Searched refs:distributeLoopOverInnerLoops (Results 1 – 5 of 5) sorted by relevance
294 static std::vector<ForPtr> distributeLoopOverInnerLoops(const ForPtr& loop);
345 l.distributeLoopOverInnerLoops(loop); in loopnestRandomization()
592 return LoopNest::distributeLoopOverInnerLoops(f); in initTensorExprBindings()
1742 std::vector<ForPtr> LoopNest::distributeLoopOverInnerLoops(const ForPtr& loop) { in distributeLoopOverInnerLoops() function in torch::jit::tensorexpr::LoopNest1751 auto result = distributeLoopOverInnerLoops(loop); in distributeLoopAndParentsOverInnerLoops()
5206 auto new_loops = LoopNest::distributeLoopOverInnerLoops(forI); in TEST()