Searched refs:PyFunctionPreHook (Results 1 – 3 of 3) sorted by relevance
127 PyFunctionPreHook::PyFunctionPreHook(PyObject* dict) : dict(dict) { in PyFunctionPreHook() function in torch::autograd::PyFunctionPreHook132 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()
22 struct PyFunctionPreHook : public FunctionPreHook { struct23 PyFunctionPreHook(PyObject* dict);24 ~PyFunctionPreHook() override;
101 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()