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