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
#pragma once
 
#include <THC/THCTensorCopy.h>
 
template <typename ScalarTypeDst, typename ScalarTypeSrc>
void THC_copyTensor(THCState* state, THCTensor* dst, THCTensor* src);
 
template <typename ScalarType>
THCTensor *THCTensor_newClone(THCState *state, THCTensor *self);
 
template <typename ScalarType>
THCTensor *THCTensor_newContiguous(THCState *state, THCTensor *self);
 
template <typename ScalarType>
void THCTensor_freeCopyTo(THCState *state, THCTensor *self, THCTensor *dst);
 
template <typename ScalarType>
void THCTensor_copyIgnoringOverlaps(THCState* state, THCTensor* dst, THCTensor* src);