Home
last modified time | relevance | path

Searched refs:forward_hooks_ (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/aten/src/ATen/core/
Dclass_type.cpp25 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()
Dclass_type.h426 std::vector<torch::jit::Function*> forward_hooks_; member