Searched refs:PyFunctionPostHook (Results 1 – 3 of 3) sorted by relevance
150 PyFunctionPostHook::PyFunctionPostHook(PyObject* dict) : dict(dict) { in PyFunctionPostHook() function in torch::autograd::PyFunctionPostHook155 PyFunctionPostHook::~PyFunctionPostHook() { in ~PyFunctionPostHook()163 auto PyFunctionPostHook::operator()( in operator ()()196 void PyFunctionPostHook::compiled_args(CompiledNodeArgs& args) { in compiled_args()
30 struct PyFunctionPostHook : public FunctionPostHook { struct31 PyFunctionPostHook(PyObject* dict);32 ~PyFunctionPostHook() override;
106 if (auto pyhook = dynamic_cast<PyFunctionPostHook*>(hook.get())) { in THPCppFunction_traverse()357 if (auto pyhook = dynamic_cast<PyFunctionPostHook*>(hook.get())) { in registerFunctionHook()368 fn.add_post_hook(std::make_unique<PyFunctionPostHook>(dict)); in registerFunctionHook()