model/pvcnn/modules/functional/src/ball_query/ball_query.hpp

#ifndef _BALL_QUERY_HPP
#define _BALL_QUERY_HPP

#include <torch/extension.h>

at::Tensor ball_query_forward(at::Tensor centers_coords,
                              at::Tensor points_coords, const float radius,
                              const int num_neighbors);

#endif