/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/saved_variable_hooks.h (272B)
#pragma once
#include
namespace torch { namespace autograd {
struct TORCH_API SavedVariableHooks {
virtual void call_pack_hook(const at::Tensor &tensor) = 0;
virtual at::Tensor call_unpack_hook() = 0;
virtual ~SavedVariableHooks() = default;
};
}}