1 #pragma once 2 3 #include <ATen/core/Tensor.h> 4 5 namespace torch::autograd { 6 7 struct TORCH_API SavedVariableHooks { 8 virtual void call_pack_hook(const at::Tensor& tensor) = 0; 9 virtual at::Tensor call_unpack_hook() = 0; 10 virtual ~SavedVariableHooks() = default; 11 }; 12 13 } // namespace torch::autograd 14