Files
Pointnet2_PyTorch/utils/cinclude/group_points_wrapper.h
T
2017-12-26 18:43:17 -05:00

9 lines
358 B
C

int group_points_wrapper(int b, int n, int c, int npoints, int nsample,
THCudaTensor *points_tensor,
THCudaIntTensor *idx_tensor, THCudaTensor *out);
int group_points_grad_wrapper(int b, int n, int c, int npoints, int nsample,
THCudaTensor *grad_out_tensor,
THCudaIntTensor *idx_tensor,
THCudaTensor *grad_points_tensor);