Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
cpu.h 714 Bytes
Newer Older
rusty1s's avatar
first c try  
rusty1s committed
1
2
3
4
5
6
7 
void scatter_add_Float (THFloatTensor  *output, THLongTensor *index, THFloatTensor  *input, int dim);
void scatter_add_Double(THDoubleTensor *output, THLongTensor *index, THDoubleTensor *input, int dim);
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);