1 | #ifndef THP_STORAGE_INC |
---|---|
2 | #define THP_STORAGE_INC |
3 | |
4 | #include <torch/csrc/Types.h> |
5 | |
6 | #define THPStorageStr "torch.UntypedStorage" |
7 | |
8 | struct THPStorage { |
9 | PyObject_HEAD c10::StorageImpl* cdata; |
10 | }; |
11 | |
12 | TORCH_PYTHON_API PyObject* THPStorage_New( |
13 | c10::intrusive_ptr<c10::StorageImpl> ptr); |
14 | extern PyObject* THPStorageClass; |
15 | |
16 | bool THPStorage_init(PyObject* module); |
17 | void THPStorage_postInit(PyObject* module); |
18 | |
19 | extern PyTypeObject THPStorageType; |
20 | |
21 | #endif |
22 |