10 #include "torch/script.h"
12 template <
class TIndex,
class TAttr>
15 const torch::Tensor& inp_neighbors_index,
16 const torch::Tensor& inp_neighbors_row_splits,
17 const torch::Tensor& inp_neighbors_attributes);
19 #ifdef BUILD_CUDA_MODULE
20 template <
class TIndex,
class TAttr>
21 std::tuple<torch::Tensor, torch::Tensor, torch::Tensor> InvertNeighborsListCUDA(
23 const torch::Tensor& inp_neighbors_index,
24 const torch::Tensor& inp_neighbors_row_splits,
25 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:16