/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/autograd
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/autograd/python_hook.h (700B)
#pragma once
#include
#include
#include
namespace torch { namespace autograd {
struct PyFunctionPreHook : public FunctionPreHook {
PyFunctionPreHook(PyObject* dict, int value_idx);
~PyFunctionPreHook() override;
variable_list operator()(const variable_list& values) override;
PyObject* dict;
int value_idx;
};
struct PyFunctionPostHook : public FunctionPostHook {
PyFunctionPostHook(PyObject* dict);
~PyFunctionPostHook() override;
variable_list operator()(const variable_list& outputs, const variable_list& inputs) override;
PyObject* dict;
};
}} // namespace torch::autograd