Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
THGreedy.h 819 Bytes
Newer Older
rusty1s's avatar
greedy cpu without degree computation: results in O(E) runtime :(  
rusty1s committed
1 
void THGreedy(THLongTensor *cluster, THLongTensor *row, THLongTensor *col);
rusty1s's avatar
added aten  
rusty1s committed
2
rusty1s's avatar
greedy cpu without degree computation: results in O(E) runtime :(  
rusty1s committed
3
4
5
6
7
8
9 
void   THByteGreedy(THLongTensor *cluster, THLongTensor *row, THLongTensor *col,   THByteTensor *weight);
void   THCharGreedy(THLongTensor *cluster, THLongTensor *row, THLongTensor *col,   THCharTensor *weight);
void  THShortGreedy(THLongTensor *cluster, THLongTensor *row, THLongTensor *col,  THShortTensor *weight);
void    THIntGreedy(THLongTensor *cluster, THLongTensor *row, THLongTensor *col,    THIntTensor *weight);
void   THLongGreedy(THLongTensor *cluster, THLongTensor *row, THLongTensor *col,   THLongTensor *weight);
void  THFloatGreedy(THLongTensor *cluster, THLongTensor *row, THLongTensor *col,  THFloatTensor *weight);
void THDoubleGreedy(THLongTensor *cluster, THLongTensor *row, THLongTensor *col, THDoubleTensor *weight);