Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
THGreedy.h 1.78 KB
Newer Older
rusty1s's avatar
added aten  
rusty1s committed
1
2
3 
void THGreedy_cluster(THLongTensor *cluster,
                      THLongTensor *row,
                      THLongTensor *col,
rusty1s's avatar
new tensor library  
rusty1s committed
4 
                      THLongTensor *deg);
rusty1s's avatar
added aten  
rusty1s committed
5
6
7
8
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
36
37
38
39
40
41
42
43
44
45
46 

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);