Searched refs:forward_hooks_ (Results 1 – 2 of 2) sorted by relevance
25 return forward_hooks_; in getForwardHooks()37 forward_hooks_.emplace_back(hook_ptr); in addForwardHook()50 for (const auto& hook : forward_hooks_) { in findForwardHook()102 const std::string& hook_name = forward_hooks_[hook_idx]->name(); in getForwardHookErrorMessage()110 : forward_hooks_[hook_idx - 1]->getSchema().returns()[0]; in getForwardHookErrorMessage()292 const torch::jit::Function* hook = forward_hooks_[hook_idx]; in checkForwardHookSchema()315 : forward_hooks_[hook_idx - 1]->getSchema().returns()[0]; in checkForwardHookSchema()
426 std::vector<torch::jit::Function*> forward_hooks_; member