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