Searched refs:forward_pre_hooks_ (Results 1 – 2 of 2) sorted by relevance
29 return forward_pre_hooks_; in getForwardPreHooks()33 forward_pre_hooks_.emplace_back(pre_hook_ptr); in addForwardPreHook()41 for (const auto& pre_hook : forward_pre_hooks_) { in findForwardPreHook()74 const std::string& pre_hook_name = forward_pre_hooks_[pre_hook_idx]->name(); in getForwardPreHookErrorMessage()195 const torch::jit::Function* pre_hook = forward_pre_hooks_[pre_hook_idx]; in checkForwardPreHookSchema()
427 std::vector<torch::jit::Function*> forward_pre_hooks_; member