Searched refs:pre_hook_ptr (Results 1 – 2 of 2) sorted by relevance
32 void ClassType::addForwardPreHook(torch::jit::Function* pre_hook_ptr) { in addForwardPreHook() argument33 forward_pre_hooks_.emplace_back(pre_hook_ptr); in addForwardPreHook()
336 void addForwardPreHook(torch::jit::Function* pre_hook_ptr);