Searched refs:getEnclosingLoopNest (Results 1 – 3 of 3) sorted by relevance
77 static std::vector<ForPtr> getEnclosingLoopNest(const StmtPtr& st);
1802 auto outer_loops = LoopNest::getEnclosingLoopNest(loop); in hasLoopCarriedDependence()2509 auto loops = LoopNest::getEnclosingLoopNest(parent); in compressBuffer()2661 std::vector<ForPtr> LoopNest::getEnclosingLoopNest(const StmtPtr& st) { in getEnclosingLoopNest() function in torch::jit::tensorexpr::LoopNest2693 loopnests.emplace_back(LoopNest::getEnclosingLoopNest(w)); in getAllLoopNestsWritingToBuf()
497 return self.getEnclosingLoopNest(s); in initTensorExprBindings()