Home
last modified time | relevance | path

Searched refs:PyFunctionPostHook (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dpython_hook.cpp150 PyFunctionPostHook::PyFunctionPostHook(PyObject* dict) : dict(dict) { in PyFunctionPostHook() function in torch::autograd::PyFunctionPostHook
155 PyFunctionPostHook::~PyFunctionPostHook() { in ~PyFunctionPostHook()
163 auto PyFunctionPostHook::operator()( in operator ()()
196 void PyFunctionPostHook::compiled_args(CompiledNodeArgs& args) { in compiled_args()
H A Dpython_hook.h30 struct PyFunctionPostHook : public FunctionPostHook { struct
31 PyFunctionPostHook(PyObject* dict);
32 ~PyFunctionPostHook() override;
H A Dpython_cpp_function.cpp106 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()