kernel.h 6.41 KB
Newer Older
1
2
3
4
#ifdef __cplusplus
extern "C" {
#endif

rusty1s's avatar
typo  
rusty1s committed
5
void scatter_mul_kernel_Float (THCState *state, int dim,       THCudaTensor *output, THCudaLongTensor *index,       THCudaTensor *input);
rusty1s's avatar
rusty1s committed
6
void scatter_mul_kernel_Double(THCState *state, int dim, THCudaDoubleTensor *output, THCudaLongTensor *index, THCudaDoubleTensor *input);
rusty1s's avatar
typo  
rusty1s committed
7
8
9
10
11
void scatter_mul_kernel_Byte  (THCState *state, int dim,   THCudaByteTensor *output, THCudaLongTensor *index,   THCudaByteTensor *input);
void scatter_mul_kernel_Char  (THCState *state, int dim,   THCudaCharTensor *output, THCudaLongTensor *index,   THCudaCharTensor *input);
void scatter_mul_kernel_Short (THCState *state, int dim,  THCudaShortTensor *output, THCudaLongTensor *index,  THCudaShortTensor *input);
void scatter_mul_kernel_Int   (THCState *state, int dim,    THCudaIntTensor *output, THCudaLongTensor *index,    THCudaIntTensor *input);
void scatter_mul_kernel_Long  (THCState *state, int dim,   THCudaLongTensor *output, THCudaLongTensor *index,   THCudaLongTensor *input);
rusty1s's avatar
rusty1s committed
12

rusty1s's avatar
typo  
rusty1s committed
13
void scatter_div_kernel_Float (THCState *state, int dim,       THCudaTensor *output, THCudaLongTensor *index,       THCudaTensor *input);
rusty1s's avatar
rusty1s committed
14
void scatter_div_kernel_Double(THCState *state, int dim, THCudaDoubleTensor *output, THCudaLongTensor *index, THCudaDoubleTensor *input);
rusty1s's avatar
typo  
rusty1s committed
15
16
17
18
19
void scatter_div_kernel_Byte  (THCState *state, int dim,   THCudaByteTensor *output, THCudaLongTensor *index,   THCudaByteTensor *input);
void scatter_div_kernel_Char  (THCState *state, int dim,   THCudaCharTensor *output, THCudaLongTensor *index,   THCudaCharTensor *input);
void scatter_div_kernel_Short (THCState *state, int dim,  THCudaShortTensor *output, THCudaLongTensor *index,  THCudaShortTensor *input);
void scatter_div_kernel_Int   (THCState *state, int dim,    THCudaIntTensor *output, THCudaLongTensor *index,    THCudaIntTensor *input);
void scatter_div_kernel_Long  (THCState *state, int dim,   THCudaLongTensor *output, THCudaLongTensor *index,   THCudaLongTensor *input);
rusty1s's avatar
rusty1s committed
20

rusty1s's avatar
typo  
rusty1s committed
21
void scatter_mean_kernel_Float (THCState *state, int dim,       THCudaTensor *output, THCudaLongTensor *index,       THCudaTensor *input,       THCudaTensor *count);
rusty1s's avatar
rusty1s committed
22
void scatter_mean_kernel_Double(THCState *state, int dim, THCudaDoubleTensor *output, THCudaLongTensor *index, THCudaDoubleTensor *input, THCudaDoubleTensor *count);
rusty1s's avatar
typo  
rusty1s committed
23
24
25
26
27
void scatter_mean_kernel_Byte  (THCState *state, int dim,   THCudaByteTensor *output, THCudaLongTensor *index,   THCudaByteTensor *input,   THCudaByteTensor *count);
void scatter_mean_kernel_Char  (THCState *state, int dim,   THCudaCharTensor *output, THCudaLongTensor *index,   THCudaCharTensor *input,   THCudaCharTensor *count);
void scatter_mean_kernel_Short (THCState *state, int dim,  THCudaShortTensor *output, THCudaLongTensor *index,  THCudaShortTensor *input,  THCudaShortTensor *count);
void scatter_mean_kernel_Int   (THCState *state, int dim,    THCudaIntTensor *output, THCudaLongTensor *index,    THCudaIntTensor *input,    THCudaIntTensor *count);
void scatter_mean_kernel_Long  (THCState *state, int dim,   THCudaLongTensor *output, THCudaLongTensor *index,   THCudaLongTensor *input,    THCudaLongTensor *count);
rusty1s's avatar
rusty1s committed
28

rusty1s's avatar
typo  
rusty1s committed
29
void scatter_max_kernel_Float (THCState *state, int dim,       THCudaTensor *output, THCudaLongTensor *index,       THCudaTensor *input, THCudaLongTensor *arg);
rusty1s's avatar
rusty1s committed
30
void scatter_max_kernel_Double(THCState *state, int dim, THCudaDoubleTensor *output, THCudaLongTensor *index, THCudaDoubleTensor *input, THCudaLongTensor *arg);
rusty1s's avatar
typo  
rusty1s committed
31
32
33
34
35
void scatter_max_kernel_Byte  (THCState *state, int dim,   THCudaByteTensor *output, THCudaLongTensor *index,   THCudaByteTensor *input, THCudaLongTensor *arg);
void scatter_max_kernel_Char  (THCState *state, int dim,   THCudaCharTensor *output, THCudaLongTensor *index,   THCudaCharTensor *input, THCudaLongTensor *arg);
void scatter_max_kernel_Short (THCState *state, int dim,  THCudaShortTensor *output, THCudaLongTensor *index,  THCudaShortTensor *input, THCudaLongTensor *arg);
void scatter_max_kernel_Int   (THCState *state, int dim,    THCudaIntTensor *output, THCudaLongTensor *index,    THCudaIntTensor *input, THCudaLongTensor *arg);
void scatter_max_kernel_Long  (THCState *state, int dim,   THCudaLongTensor *output, THCudaLongTensor *index,   THCudaLongTensor *input, THCudaLongTensor *arg);
rusty1s's avatar
rusty1s committed
36

rusty1s's avatar
typo  
rusty1s committed
37
void scatter_min_kernel_Float (THCState *state, int dim,       THCudaTensor *output, THCudaLongTensor *index,       THCudaTensor *input, THCudaLongTensor *arg);
rusty1s's avatar
rusty1s committed
38
void scatter_min_kernel_Double(THCState *state, int dim, THCudaDoubleTensor *output, THCudaLongTensor *index, THCudaDoubleTensor *input, THCudaLongTensor *arg);
rusty1s's avatar
typo  
rusty1s committed
39
40
41
42
43
void scatter_min_kernel_Byte  (THCState *state, int dim,   THCudaByteTensor *output, THCudaLongTensor *index,   THCudaByteTensor *input, THCudaLongTensor *arg);
void scatter_min_kernel_Char  (THCState *state, int dim,   THCudaCharTensor *output, THCudaLongTensor *index,   THCudaCharTensor *input, THCudaLongTensor *arg);
void scatter_min_kernel_Short (THCState *state, int dim,  THCudaShortTensor *output, THCudaLongTensor *index,  THCudaShortTensor *input, THCudaLongTensor *arg);
void scatter_min_kernel_Int   (THCState *state, int dim,    THCudaIntTensor *output, THCudaLongTensor *index,    THCudaIntTensor *input, THCudaLongTensor *arg);
void scatter_min_kernel_Long  (THCState *state, int dim,   THCudaLongTensor *output, THCudaLongTensor *index,   THCudaLongTensor *input, THCudaLongTensor *arg);
rusty1s's avatar
rusty1s committed
44

rusty1s's avatar
typo  
rusty1s committed
45
void index_backward_kernel_Float (THCState *state, int dim,       THCudaTensor *output, THCudaLongTensor *index,       THCudaTensor *grad, THCudaLongTensor *arg);
rusty1s's avatar
rusty1s committed
46
void index_backward_kernel_Double(THCState *state, int dim, THCudaDoubleTensor *output, THCudaLongTensor *index, THCudaDoubleTensor *grad, THCudaLongTensor *arg);
rusty1s's avatar
typo  
rusty1s committed
47
48
49
50
51
void index_backward_kernel_Byte  (THCState *state, int dim,   THCudaByteTensor *output, THCudaLongTensor *index,   THCudaByteTensor *grad, THCudaLongTensor *arg);
void index_backward_kernel_Char  (THCState *state, int dim,   THCudaCharTensor *output, THCudaLongTensor *index,   THCudaCharTensor *grad, THCudaLongTensor *arg);
void index_backward_kernel_Short (THCState *state, int dim,  THCudaShortTensor *output, THCudaLongTensor *index,  THCudaShortTensor *grad, THCudaLongTensor *arg);
void index_backward_kernel_Int   (THCState *state, int dim,    THCudaIntTensor *output, THCudaLongTensor *index,    THCudaIntTensor *grad, THCudaLongTensor *arg);
void index_backward_kernel_Long  (THCState *state, int dim,   THCudaLongTensor *output, THCudaLongTensor *index,   THCudaLongTensor *grad, THCudaLongTensor *arg);
52
53
54
55

#ifdef __cplusplus
}
#endif