1#pragma once
2
3#include <ATen/core/Tensor.h>
4
5namespace torch {
6namespace autograd {
7
8struct 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