reid from https://github.com/michuanhaohao/reid-strong-baseline
554325746@qq.com
2020-02-05 7c6a958f00e327a8991f48093456de7cb84cf918
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);