Home
last modified time | relevance | path

Searched defs:getLoopStmtsFor (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dloopnest.cpp2597 std::vector<ForPtr> LoopNest::getLoopStmtsFor(const Tensor& t) const { in getLoopStmtsFor() function in torch::jit::tensorexpr::LoopNest
2602 std::vector<ForPtr> LoopNest::getLoopStmtsFor(const BufPtr& buf) const { in getLoopStmtsFor() function in torch::jit::tensorexpr::LoopNest
2607 std::vector<ForPtr> LoopNest::getLoopStmtsFor(StmtPtr s) const { in getLoopStmtsFor() function in torch::jit::tensorexpr::LoopNest