Searched refs:getLoopStmtsInLoopNest (Results 1 – 3 of 3) sorted by relevance
479 static std::vector<ForPtr> getLoopStmtsInLoopNest(
529 return LoopNest::getLoopStmtsInLoopNest(f, num); in initTensorExprBindings()
2377 std::vector<ForPtr> LoopNest::getLoopStmtsInLoopNest( in getLoopStmtsInLoopNest() function in torch::jit::tensorexpr::LoopNest2434 auto normalized_loops = getLoopStmtsInLoopNest(loops.front(), loops.size()); in flatten()