THCGreedy.h 392 Bytes
Newer Older
rusty1s's avatar
rusty1s committed
1
2
3
4
5
6
7
8
9
10
11
12
13
#ifndef THC_GENERIC_FILE
#define THC_GENERIC_FILE "generic/THCGreedy.h"
#else

void THCGreedy_(cluster)(THCState *state,
                         THCudaLongTensor *cluster,
                         THCudaLongTensor *row,
                         THCudaLongTensor *col,
                         THCudaLongTensor *deg,
                         THCTensor *weight);

#endif  // THC_GENERIC_FILE