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