1 | #pragma once |
---|---|
2 | |
3 | #include <c10/core/StorageImpl.h> |
4 | #include <c10/core/TensorImpl.h> |
5 | #include <torch/csrc/Types.h> |
6 | #include <torch/csrc/python_headers.h> |
7 | |
8 | namespace torch { |
9 | |
10 | struct THPVoidTensor { |
11 | PyObject_HEAD c10::TensorImpl* cdata; |
12 | char device_type; |
13 | char data_type; |
14 | }; |
15 | |
16 | struct THPVoidStorage { |
17 | PyObject_HEAD c10::StorageImpl* cdata; |
18 | }; |
19 | |
20 | } // namespace torch |
21 |