mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
26 lines
730 B
C
26 lines
730 B
C
#ifndef THP_MODULE_INC
|
|
#define THP_MODULE_INC
|
|
|
|
extern THPGenerator *THPDefaultGenerator;
|
|
bool THPModule_tensorCopy(PyObject *dst, PyObject *src);
|
|
|
|
#define STATELESS_ATTRIBUTE_NAME "_torch"
|
|
|
|
extern PyObject *THPDoubleStorageClass;
|
|
extern PyObject *THPFloatStorageClass;
|
|
extern PyObject *THPLongStorageClass;
|
|
extern PyObject *THPIntStorageClass;
|
|
extern PyObject *THPShortStorageClass;
|
|
extern PyObject *THPCharStorageClass;
|
|
extern PyObject *THPByteStorageClass;
|
|
|
|
extern PyObject *THPDoubleTensorClass;
|
|
extern PyObject *THPFloatTensorClass;
|
|
extern PyObject *THPLongTensorClass;
|
|
extern PyObject *THPIntTensorClass;
|
|
extern PyObject *THPShortTensorClass;
|
|
extern PyObject *THPCharTensorClass;
|
|
extern PyObject *THPByteTensorClass;
|
|
|
|
#endif
|