1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26
|
#pragma once
#include <torch/csrc/python_headers.h>
#include <ATen/ATen.h>
struct THPDTypeInfo {
PyObject_HEAD at::ScalarType type;
};
struct THPFInfo : THPDTypeInfo {};
struct THPIInfo : THPDTypeInfo {};
extern PyTypeObject THPFInfoType;
extern PyTypeObject THPIInfoType;
inline bool THPFInfo_Check(PyObject* obj) {
return Py_TYPE(obj) == &THPFInfoType;
}
inline bool THPIInfo_Check(PyObject* obj) {
return Py_TYPE(obj) == &THPIInfoType;
}
void THPDTypeInfo_init(PyObject* module);
|