Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
THCWeighting.h 278 Bytes
Newer Older
rusty1s's avatar
weigthing forward (cpu+gpu)  
rusty1s committed
1
2 
#ifndef THC_WEIGHTING_INC
#define THC_WEIGHTING_INC
rusty1s's avatar
added conv kernel headers  
rusty1s committed
3
4
5
6
7
8
9 

#include <THC/THC.h>

#ifdef __cplusplus
extern "C" {
#endif  // __cplusplus
rusty1s's avatar
weigthing forward (cpu+gpu)  
rusty1s committed
10 
#include "generic/THCWeighting.h"
rusty1s's avatar
added conv kernel headers  
rusty1s committed
11
12
13
14
15
16 
#include "THC/THCGenerateFloatTypes.h"

#ifdef __cplusplus
}
#endif  // __cplusplus
rusty1s's avatar
weigthing forward (cpu+gpu)  
rusty1s committed
17 
#endif // THC_WEIGHTING_INC