Skip to content

GitLab

"examples/nas/oneshot/spos/architecture_final.json" did not exist on "4f3ee9cb3f2d39dc3a105c928cef62d9520073af"
Switch branch/tag
Find file Normal viewHistoryPermalink
THGreedy.h 1.19 KB
Newer Older
rusty1s's avatar
new tensor library  
rusty1s committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16 
void THGreedy_cluster(THLongTensor *cluster, THLongTensor *row, THLongTensor *col,
                      THLongTensor *deg);
void THByteGreedy_cluster(THLongTensor *cluster, THLongTensor *row, THLongTensor *col,
                          THLongTensor *deg, THByteTensor *weight);
void THCharGreedy_cluster(THLongTensor *cluster, THLongTensor *row, THLongTensor *col,
                          THLongTensor *deg, THCharTensor *weight);
void THShortGreedy_cluster(THLongTensor *cluster, THLongTensor *row, THLongTensor *col,
                           THLongTensor *deg, THShortTensor *weight);
void THIntGreedy_cluster(THLongTensor *cluster, THLongTensor *row, THLongTensor *col,
                         THLongTensor *deg, THIntTensor *weight);
void THLongGreedy_cluster(THLongTensor *cluster, THLongTensor *row, THLongTensor *col,
                          THLongTensor *deg, THLongTensor *weight);
void THFloatGreedy_cluster(THLongTensor *cluster, THLongTensor *row, THLongTensor *col,
                           THLongTensor *deg, THFloatTensor *weight);
void THDoubleGreedy_cluster(THLongTensor *cluster, THLongTensor *row, THLongTensor *col,
                            THLongTensor *deg, THDoubleTensor *weight);