File size: 99 Bytes
29858c0
 
1
2
3
int knn(THCudaTensor *ref_tensor, THCudaTensor *query_tensor,

    THCudaLongTensor *idx_tensor);