| 548 | |
| 549 | |
| 550 | torch::Tensor keep_min_k_values( |
| 551 | torch::Tensor values, // (b, n), float |
| 552 | int64_t k |
| 553 | ) { |
| 554 | // return values; |
| 555 | return keep_min_k_values_cuda( |
| 556 | values, // (b, n, |
| 557 | k |
| 558 | ); |
| 559 | } |
| 560 | |
| 561 | |
| 562 | // Find the k nearest points within `radius` of a ray, ie, the vertical distance from the point to ray <= radius. |
nothing calls this directly
no outgoing calls
no test coverage detected