reid from https://github.com/michuanhaohao/reid-strong-baseline
zhangmeng
2020-01-10 c3765bd24fe73747688a0ec2a550f219c9acb384
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
#ifndef THC_TENSOR_INC
#define THC_TENSOR_INC
 
#include <TH/THTensor.h>
#include <THC/THCStorage.h>
#include <THC/THCGeneral.h>
 
#define THCTensor_(NAME)   TH_CONCAT_4(TH,CReal,Tensor_,NAME)
 
#define THC_DESC_BUFF_LEN 64
 
typedef struct THC_CLASS THCDescBuff
{
    char str[THC_DESC_BUFF_LEN];
} THCDescBuff;
 
#include <THC/generic/THCTensor.h>
#include <THC/THCGenerateAllTypes.h>
 
#include <THC/generic/THCTensor.h>
#include <THC/THCGenerateBoolType.h>
 
#include <THC/generic/THCTensor.h>
#include <THC/THCGenerateBFloat16Type.h>
 
#endif