Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
THCCGreedy.h 1.25 KB
Newer Older
rusty1s's avatar
thcc wrapper methods  
rusty1s committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17 
void THCCGreedy(THCudaLongTensor *cluster, THCudaLongTensor *row, THCudaLongTensor *col,
                THCudaLongTensor *deg);

void THCCByteGreedy(THCudaLongTensor *cluster, THCudaLongTensor *row, THCudaLongTensor *col,
                    THCudaLongTensor *deg, THCudaByteTensor *weight);
void THCCCharGreedy(THCudaLongTensor *cluster, THCudaLongTensor *row, THCudaLongTensor *col,
                    THCudaLongTensor *deg, THCudaCharTensor *weight);
void THCCShortGreedy(THCudaLongTensor *cluster, THCudaLongTensor *row, THCudaLongTensor *col,
                     THCudaLongTensor *deg, THCudaShortTensor *weight);
void THCCIntGreedy(THCudaLongTensor *cluster, THCudaLongTensor *row, THCudaLongTensor *col,
                   THCudaLongTensor *deg, THCudaIntTensor *weight);
void THCCLongGreedy(THCudaLongTensor *cluster, THCudaLongTensor *row, THCudaLongTensor *col,
                    THCudaLongTensor *deg, THCudaLongTensor *weight);
void THCCFloatGreedy(THCudaLongTensor *cluster, THCudaLongTensor *row, THCudaLongTensor *col,
                     THCudaLongTensor *deg, THCudaTensor *weight);
void THCCDoubleGreedy(THCudaLongTensor *cluster, THCudaLongTensor *row, THCudaLongTensor *col,
                      THCudaLongTensor *deg, THCudaDoubleTensor *weight);