mirror of
https://github.com/wassname/Pointnet2_PyTorch.git
synced 2026-06-27 16:00:07 +08:00
erikwijmans
15 lines
519 B
C
15 lines
519 B
C
|
|
int gather_points_wrapper(int b, int c, int n, int npoints,
|
|
THCudaTensor *points_tensor,
|
|
THCudaIntTensor *idx_tensor,
|
|
THCudaTensor *out_tensor);
|
|
int gather_points_grad_wrapper(int b, int c, int n, int npoints,
|
|
THCudaTensor *grad_out_tensor,
|
|
THCudaIntTensor *idx_tensor,
|
|
THCudaTensor *grad_points_tensor);
|
|
|
|
int furthest_point_sampling_wrapper(int b, int n, int m,
|
|
THCudaTensor *points_tensor,
|
|
THCudaTensor *temp_tensor,
|
|
THCudaIntTensor *idx_tensor);
|