Searched defs:LoopsPeeler (Results 1 – 1 of 1) sorted by relevance
18 struct TORCH_API LoopsPeeler { struct20 : callback_(std::move(callback)), num_iterations_(num_iterations) {} in callback_()29 std::function<bool(Node* n)> callback_ = nullptr;30 Node* in_loop_ = nullptr;31 std::list<Node*> loops_to_peel_;32 size_t num_iterations_ = 1;