reid from https://github.com/michuanhaohao/reid-strong-baseline
zhangmeng
2020-01-20 455a7bb3e42582a62a02d7baf1b1d4495bf6107c
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
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
#ifndef TH_GENERIC_FILE
#define TH_GENERIC_FILE "TH/generic/THTensorConv.h"
#else
 
TH_API void THTensor_(validXCorr2Dptr)(scalar_t *r_,
                                    scalar_t alpha,
                                    scalar_t *t_, int64_t ir, int64_t ic,
                                    scalar_t *k_, int64_t kr, int64_t kc,
                                    int64_t sr, int64_t sc);
 
TH_API void THTensor_(validConv2Dptr)(scalar_t *r_,
                                   scalar_t alpha,
                                   scalar_t *t_, int64_t ir, int64_t ic,
                                   scalar_t *k_, int64_t kr, int64_t kc,
                                   int64_t sr, int64_t sc);
 
TH_API void THTensor_(fullXCorr2Dptr)(scalar_t *r_,
                                   scalar_t alpha,
                                   scalar_t *t_, int64_t ir, int64_t ic,
                                   scalar_t *k_, int64_t kr, int64_t kc,
                                   int64_t sr, int64_t sc);
 
TH_API void THTensor_(fullConv2Dptr)(scalar_t *r_,
                                  scalar_t alpha,
                                  scalar_t *t_, int64_t ir, int64_t ic,
                                  scalar_t *k_, int64_t kr, int64_t kc,
                                  int64_t sr, int64_t sc);
 
TH_API void THTensor_(validXCorr2DRevptr)(scalar_t *r_,
                                       scalar_t alpha,
                                       scalar_t *t_, int64_t ir, int64_t ic,
                                       scalar_t *k_, int64_t kr, int64_t kc,
                                       int64_t sr, int64_t sc);
 
TH_API void THTensor_(conv2DRevger)(THTensor *r_, scalar_t beta, scalar_t alpha, THTensor *t_, THTensor *k_, int64_t srow, int64_t scol);
TH_API void THTensor_(conv2DRevgerm)(THTensor *r_, scalar_t beta, scalar_t alpha, THTensor *t_, THTensor *k_, int64_t srow, int64_t scol);
TH_API void THTensor_(conv2Dger)(THTensor *r_, scalar_t beta, scalar_t alpha, THTensor *t_, THTensor *k_, int64_t srow, int64_t scol, const char *vf, const char *xc);
TH_API void THTensor_(conv2Dmv)(THTensor *r_, scalar_t beta, scalar_t alpha, THTensor *t_, THTensor *k_, int64_t srow, int64_t scol, const char *vf, const char *xc);
TH_API void THTensor_(conv2Dmm)(THTensor *r_, scalar_t beta, scalar_t alpha, THTensor *t_, THTensor *k_, int64_t srow, int64_t scol, const char *vf, const char *xc);
TH_API void THTensor_(conv2Dmul)(THTensor *r_, scalar_t beta, scalar_t alpha, THTensor *t_, THTensor *k_, int64_t srow, int64_t scol, const char *vf, const char *xc);
TH_API void THTensor_(conv2Dcmul)(THTensor *r_, scalar_t beta, scalar_t alpha, THTensor *t_, THTensor *k_, int64_t srow, int64_t scol, const char *vf, const char *xc);
 
TH_API void THTensor_(validXCorr3Dptr)(scalar_t *r_,
                                    scalar_t alpha,
                                    scalar_t *t_, int64_t it, int64_t ir, int64_t ic,
                                    scalar_t *k_, int64_t kt, int64_t kr, int64_t kc,
                                    int64_t st, int64_t sr, int64_t sc);
 
TH_API void THTensor_(validConv3Dptr)(scalar_t *r_,
                                   scalar_t alpha,
                                   scalar_t *t_, int64_t it, int64_t ir, int64_t ic,
                                   scalar_t *k_, int64_t kt, int64_t kr, int64_t kc,
                                   int64_t st, int64_t sr, int64_t sc);
 
TH_API void THTensor_(fullXCorr3Dptr)(scalar_t *r_,
                                   scalar_t alpha,
                                   scalar_t *t_, int64_t it, int64_t ir, int64_t ic,
                                   scalar_t *k_, int64_t kt, int64_t kr, int64_t kc,
                                   int64_t st, int64_t sr, int64_t sc);
 
TH_API void THTensor_(fullConv3Dptr)(scalar_t *r_,
                                  scalar_t alpha,
                                  scalar_t *t_, int64_t it, int64_t ir, int64_t ic,
                                  scalar_t *k_, int64_t kt, int64_t kr, int64_t kc,
                                  int64_t st, int64_t sr, int64_t sc);
 
TH_API void THTensor_(validXCorr3DRevptr)(scalar_t *r_,
                                       scalar_t alpha,
                                       scalar_t *t_, int64_t it, int64_t ir, int64_t ic,
                                       scalar_t *k_, int64_t kt, int64_t kr, int64_t kc,
                                       int64_t st, int64_t sr, int64_t sc);
 
TH_API void THTensor_(conv3DRevger)(THTensor *r_, scalar_t beta, scalar_t alpha, THTensor *t_, THTensor *k_, int64_t sdepth, int64_t srow, int64_t scol);
TH_API void THTensor_(conv3Dger)(THTensor *r_, scalar_t beta, scalar_t alpha, THTensor *t_, THTensor *k_, int64_t sdepth, int64_t srow, int64_t scol, const char *vf, const char *xc);
TH_API void THTensor_(conv3Dmv)(THTensor *r_, scalar_t beta, scalar_t alpha, THTensor *t_, THTensor *k_, int64_t sdepth, int64_t srow, int64_t scol, const char *vf, const char *xc);
TH_API void THTensor_(conv3Dmul)(THTensor *r_, scalar_t beta, scalar_t alpha, THTensor *t_, THTensor *k_, int64_t sdepth, int64_t srow, int64_t scol, const char *vf, const char *xc);
TH_API void THTensor_(conv3Dcmul)(THTensor *r_, scalar_t beta, scalar_t alpha, THTensor *t_, THTensor *k_, int64_t sdepth, int64_t srow, int64_t scol, const char *vf, const char *xc);
 
#endif