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);
|
