Searched defs:addForwardHook (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/torch/csrc/jit/frontend/ | ||
| D | concrete_module_type.cpp | 281 void ConcreteModuleTypeBuilder::addForwardHook(py::object hook) { in addForwardHook() function in torch::jit::ConcreteModuleTypeBuilder |
| /external/pytorch/aten/src/ATen/core/ | ||
| D | class_type.cpp | 36 void ClassType::addForwardHook(torch::jit::Function* hook_ptr) { in addForwardHook() function in c10::ClassType |