Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
THCGreedy.h 2.04 KB
Newer Older
rusty1s's avatar
added aten  
rusty1s committed
1
2
3
4
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 THCudaGreedyWrapper(THCudaLongTensor *cluster,
                         THCudaLongTensor *row,
                         THCudaLongTensor *col,
                         THCudaLongTensor *deg);

void THCudaByteGreedyWrapper(THCudaLongTensor *cluster,
                             THCudaLongTensor *row,
                             THCudaLongTensor *col,
                             THCudaLongTensor *deg,
                             THCudaByteTensor *weight);

void THCudaCharGreedyWrapper(THCudaLongTensor *cluster,
                             THCudaLongTensor *row,
                             THCudaLongTensor *col,
                             THCudaLongTensor *deg,
                             THCudaCharTensor *weight);

void THCudaShortGreedyWrapper(THCudaLongTensor *cluster,
                              THCudaLongTensor *row,
                              THCudaLongTensor *col,
                              THCudaLongTensor *deg,
                              THCudaShortTensor *weight);

void THCudaIntGreedyWrapper(THCudaLongTensor *cluster,
                            THCudaLongTensor *row,
                            THCudaLongTensor *col,
                            THCudaLongTensor *deg,
                            THCudaIntTensor *weight);

void THCudaLongGreedyWrapper(THCudaLongTensor *cluster,
                             THCudaLongTensor *row,
                             THCudaLongTensor *col,
                             THCudaLongTensor *deg,
                             THCudaLongTensor *weight);

void THCudaFloatGreedyWrapper(THCudaLongTensor *cluster,
                              THCudaLongTensor *row,
                              THCudaLongTensor *col,
                              THCudaLongTensor *deg,
                              THCudaTensor *weight);

void THCudaDoubleGreedyWrapper(THCudaLongTensor *cluster,
                               THCudaLongTensor *row,
                               THCudaLongTensor *col,
                               THCudaLongTensor *deg,
                               THCudaDoubleTensor *weight);