Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
serial_cpu.h 1.02 KB
Newer Older
rusty1s's avatar
fixes  
rusty1s committed
1 
void cluster_serial(THLongTensor *output, THLongTensor *row, THLongTensor *col, THLongTensor *degree);
rusty1s's avatar
rename, outsource cpu methods  
rusty1s committed
2
rusty1s's avatar
changed arg order  
rusty1s committed
3
4
5
6
7
8
9 
void cluster_serial_Float (THLongTensor *output, THLongTensor *row, THLongTensor *col, THLongTensor *degree,  THFloatTensor *weight);
void cluster_serial_Double(THLongTensor *output, THLongTensor *row, THLongTensor *col, THLongTensor *degree, THDoubleTensor *weight);
void cluster_serial_Byte  (THLongTensor *output, THLongTensor *row, THLongTensor *col, THLongTensor *degree,   THByteTensor *weight);
void cluster_serial_Char  (THLongTensor *output, THLongTensor *row, THLongTensor *col, THLongTensor *degree,   THCharTensor *weight);
void cluster_serial_Short (THLongTensor *output, THLongTensor *row, THLongTensor *col, THLongTensor *degree,  THShortTensor *weight);
void cluster_serial_Int   (THLongTensor *output, THLongTensor *row, THLongTensor *col, THLongTensor *degree,    THIntTensor *weight);
void cluster_serial_Long  (THLongTensor *output, THLongTensor *row, THLongTensor *col, THLongTensor *degree,   THLongTensor *weight);