Searched refs:getLoopAt (Results 1 – 4 of 4) sorted by relevance
70 ForPtr getLoopAt(ForPtr root, const std::vector<int>& indices) const;
517 return self.getLoopAt(std::move(root), indices); in initTensorExprBindings()
2221 ForPtr LoopNest::getLoopAt(ForPtr root, const std::vector<int>& indices) const { in getLoopAt() function in torch::jit::tensorexpr::LoopNest
638 TEST(LoopNest, getLoopAt) { in TEST() argument682 auto ret_k2 = l.getLoopAt(for_i, {0, 2}); in TEST()