Caffe2 - C++ API
A deep learning, cross platform ML framework
Storage.h
1 #ifndef THP_STORAGE_INC
2 #define THP_STORAGE_INC
3 
4 #define THPStorageStr TH_CONCAT_STRING_3(torch.,Real,Storage)
5 #define THPStorageClass TH_CONCAT_3(THP,Real,StorageClass)
6 #define THPStorage_(NAME) TH_CONCAT_4(THP,Real,Storage_,NAME)
7 
8 #define THPDoubleStorage_Check(obj) \
9  PyObject_IsInstance(obj, THPDoubleStorageClass)
10 #define THPFloatStorage_Check(obj) \
11  PyObject_IsInstance(obj, THPFloatStorageClass)
12 #define THPHalfStorage_Check(obj) \
13  PyObject_IsInstance(obj, THPFloatStorageClass)
14 #define THPLongStorage_Check(obj) \
15  PyObject_IsInstance(obj, THPLongStorageClass)
16 #define THPIntStorage_Check(obj) \
17  PyObject_IsInstance(obj, THPIntStorageClass)
18 #define THPShortStorage_Check(obj) \
19  PyObject_IsInstance(obj, THPShortStorageClass)
20 #define THPCharStorage_Check(obj) \
21  PyObject_IsInstance(obj, THPCharStorageClass)
22 #define THPByteStorage_Check(obj) \
23  PyObject_IsInstance(obj, THPByteStorageClass)
24 #define THPBoolStorage_Check(obj) \
25  PyObject_IsInstance(obj, THPBoolStorageClass)
26 
27 #define THPDoubleStorage_CData(obj) (obj)->cdata
28 #define THPFloatStorage_CData(obj) (obj)->cdata
29 #define THPHalfStorage_CData(obj) (obj)->cdata
30 #define THPLongStorage_CData(obj) (obj)->cdata
31 #define THPIntStorage_CData(obj) (obj)->cdata
32 #define THPShortStorage_CData(obj) (obj)->cdata
33 #define THPCharStorage_CData(obj) (obj)->cdata
34 #define THPByteStorage_CData(obj) (obj)->cdata
35 #define THPBoolStorage_CData(obj) (obj)->cdata
36 
37 #ifdef _THP_CORE
38 #define THPStorageType TH_CONCAT_3(THP,Real,StorageType)
39 #define THPStorageBaseStr TH_CONCAT_STRING_2(Real,StorageBase)
40 #endif
41 
42 #include <torch/csrc/generic/Storage.h>
43 #include <TH/THGenerateAllTypes.h>
44 
45 #include <torch/csrc/generic/Storage.h>
46 #include <TH/THGenerateHalfType.h>
47 
48 #include <torch/csrc/generic/Storage.h>
49 #include <TH/THGenerateBoolType.h>
50 
51 #endif