Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dpython_hook.cpp127 PyFunctionPreHook::PyFunctionPreHook(PyObject* dict) : dict(dict) { in PyFunctionPreHook() function in torch::autograd::PyFunctionPreHook
132 PyFunctionPreHook::~PyFunctionPreHook() { in ~PyFunctionPreHook()
140 auto PyFunctionPreHook::operator()(const variable_list& grad_outputs_) in operator ()()
187 void PyFunctionPreHook::compiled_args(CompiledNodeArgs& args) { in compiled_args()
H A Dpython_hook.h22 struct PyFunctionPreHook : public FunctionPreHook { struct
23 PyFunctionPreHook(PyObject* dict);
24 ~PyFunctionPreHook() override;
H A Dpython_cpp_function.cpp101 if (auto pyhook = dynamic_cast<PyFunctionPreHook*>(hook.get())) { in THPCppFunction_traverse()
380 if (auto pyhook = dynamic_cast<PyFunctionPreHook*>(hook.get())) { in registerFunctionPreHook()
391 fn.add_pre_hook(std::make_unique<PyFunctionPreHook>(dict)); in registerFunctionPreHook()