mirror of
https://github.com/wassname/Pointnet2_PyTorch.git
synced 2026-06-27 16:00:07 +08:00
erikwijmans
17 lines
611 B
C
17 lines
611 B
C
|
|
|
|
void three_nn_wrapper(int b, int n, int m, THCudaTensor *unknown_tensor,
|
|
THCudaTensor *known_tensor, THCudaTensor *dist2_tensor,
|
|
THCudaIntTensor *idx_tensor);
|
|
void three_interpolate_wrapper(int b, int c, int m, int n,
|
|
THCudaTensor *points_tensor,
|
|
THCudaIntTensor *idx_tensor,
|
|
THCudaTensor *weight_tensor,
|
|
THCudaTensor *out_tensor);
|
|
|
|
void three_interpolate_grad_wrapper(int b, int c, int n, int m,
|
|
THCudaTensor *grad_out_tensor,
|
|
THCudaIntTensor *idx_tensor,
|
|
THCudaTensor *weight_tensor,
|
|
THCudaTensor *grad_points_tensor);
|