Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dloopnest.h479 static std::vector<ForPtr> getLoopStmtsInLoopNest(
H A Dtensorexpr_init.cpp529 return LoopNest::getLoopStmtsInLoopNest(f, num); in initTensorExprBindings()
H A Dloopnest.cpp2377 std::vector<ForPtr> LoopNest::getLoopStmtsInLoopNest( in getLoopStmtsInLoopNest() function in torch::jit::tensorexpr::LoopNest
2434 auto normalized_loops = getLoopStmtsInLoopNest(loops.front(), loops.size()); in flatten()