29 #include "torch/script.h"
31 template <
class TIndex,
class TAttr>
34 const torch::Tensor& inp_neighbors_index,
35 const torch::Tensor& inp_neighbors_row_splits,
36 const torch::Tensor& inp_neighbors_attributes);
38 #ifdef BUILD_CUDA_MODULE
39 template <
class TIndex,
class TAttr>
40 std::tuple<torch::Tensor, torch::Tensor, torch::Tensor> InvertNeighborsListCUDA(
42 const torch::Tensor& inp_neighbors_index,
43 const torch::Tensor& inp_neighbors_row_splits,
44 const torch::Tensor& inp_neighbors_attributes);
std::tuple< torch::Tensor, torch::Tensor, torch::Tensor > InvertNeighborsListCPU(int64_t num_points, const torch::Tensor &inp_neighbors_index, const torch::Tensor &inp_neighbors_row_splits, const torch::Tensor &inp_neighbors_attributes)
Definition: InvertNeighborsListOpKernel.cpp:35