| 13 | } |
| 14 | |
| 15 | void grouping_backward_cuda(int m, int nsample, int c, at::Tensor grad_output_tensor, at::Tensor idx_tensor, at::Tensor grad_input_tensor) |
| 16 | { |
| 17 | const float *grad_output = grad_output_tensor.data_ptr<float>(); |
| 18 | const int *idx = idx_tensor.data_ptr<int>(); |
| 19 | float *grad_input = grad_input_tensor.data_ptr<float>(); |
| 20 | grouping_backward_cuda_launcher(m, nsample, c, grad_output, idx, grad_input); |
| 21 | } |
nothing calls this directly
no outgoing calls
no test coverage detected