Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
cpu.h 3.21 KB
Newer Older
rusty1s's avatar
first c try  
rusty1s committed
1
2 
void scatter_add_Float (THFloatTensor  *output, THLongTensor *index, THFloatTensor  *input, int dim);
void scatter_add_Double(THDoubleTensor *output, THLongTensor *index, THDoubleTensor *input, int dim);
rusty1s's avatar
long to int64_t  
rusty1s committed
3 
/* void scatter_add_Half  (THHalfTensor   *output, THLongTensor *index, THHalfTensor   *input, int dim); */
rusty1s's avatar
first c try  
rusty1s committed
4
5
6
7
8 
void scatter_add_Byte  (THByteTensor   *output, THLongTensor *index, THByteTensor   *input, int dim);
void scatter_add_Char  (THCharTensor   *output, THLongTensor *index, THCharTensor   *input, int dim);
void scatter_add_Short (THShortTensor  *output, THLongTensor *index, THShortTensor  *input, int dim);
void scatter_add_Int   (THIntTensor    *output, THLongTensor *index, THIntTensor    *input, int dim);
void scatter_add_Long  (THLongTensor   *output, THLongTensor *index, THLongTensor   *input, int dim);
rusty1s's avatar
long to int64_t  
rusty1s committed
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 

void scatter_sub_Float (THFloatTensor  *output, THLongTensor *index, THFloatTensor  *input, int dim);
void scatter_sub_Double(THDoubleTensor *output, THLongTensor *index, THDoubleTensor *input, int dim);
/* void scatter_sub_Half  (THHalfTensor   *output, THLongTensor *index, THHalfTensor   *input, int dim); */
void scatter_sub_Byte  (THByteTensor   *output, THLongTensor *index, THByteTensor   *input, int dim);
void scatter_sub_Char  (THCharTensor   *output, THLongTensor *index, THCharTensor   *input, int dim);
void scatter_sub_Short (THShortTensor  *output, THLongTensor *index, THShortTensor  *input, int dim);
void scatter_sub_Int   (THIntTensor    *output, THLongTensor *index, THIntTensor    *input, int dim);
void scatter_sub_Long  (THLongTensor   *output, THLongTensor *index, THLongTensor   *input, int dim);

void scatter_mul_Float (THFloatTensor  *output, THLongTensor *index, THFloatTensor  *input, int dim);
void scatter_mul_Double(THDoubleTensor *output, THLongTensor *index, THDoubleTensor *input, int dim);
/* void scatter_mul_Half  (THHalfTensor   *output, THLongTensor *index, THHalfTensor   *input, int dim); */
void scatter_mul_Byte  (THByteTensor   *output, THLongTensor *index, THByteTensor   *input, int dim);
void scatter_mul_Char  (THCharTensor   *output, THLongTensor *index, THCharTensor   *input, int dim);
void scatter_mul_Short (THShortTensor  *output, THLongTensor *index, THShortTensor  *input, int dim);
void scatter_mul_Int   (THIntTensor    *output, THLongTensor *index, THIntTensor    *input, int dim);
void scatter_mul_Long  (THLongTensor   *output, THLongTensor *index, THLongTensor   *input, int dim);

void scatter_div_Float (THFloatTensor  *output, THLongTensor *index, THFloatTensor  *input, int dim);
void scatter_div_Double(THDoubleTensor *output, THLongTensor *index, THDoubleTensor *input, int dim);
/* void scatter_div_Half  (THHalfTensor   *output, THLongTensor *index, THHalfTensor   *input, int dim); */
void scatter_div_Byte  (THByteTensor   *output, THLongTensor *index, THByteTensor   *input, int dim);
void scatter_div_Char  (THCharTensor   *output, THLongTensor *index, THCharTensor   *input, int dim);
void scatter_div_Short (THShortTensor  *output, THLongTensor *index, THShortTensor  *input, int dim);
void scatter_div_Int   (THIntTensor    *output, THLongTensor *index, THIntTensor    *input, int dim);
void scatter_div_Long  (THLongTensor   *output, THLongTensor *index, THLongTensor   *input, int dim);