Open3D (C++ API)  0.17.0
Loading...
Searching...
No Matches
InvertNeighborsListOpKernel.h
Go to the documentation of this file.
1// ----------------------------------------------------------------------------
2// - Open3D: www.open3d.org -
3// ----------------------------------------------------------------------------
4// Copyright (c) 2018-2023 www.open3d.org
5// SPDX-License-Identifier: MIT
6// ----------------------------------------------------------------------------
7//
8#pragma once
9
10#include "torch/script.h"
11
12template <class TIndex, class TAttr>
13std::tuple<torch::Tensor, torch::Tensor, torch::Tensor> InvertNeighborsListCPU(
14 int64_t num_points,
15 const torch::Tensor& inp_neighbors_index,
16 const torch::Tensor& inp_neighbors_row_splits,
17 const torch::Tensor& inp_neighbors_attributes);
18
19#ifdef BUILD_CUDA_MODULE
20template <class TIndex, class TAttr>
21std::tuple<torch::Tensor, torch::Tensor, torch::Tensor> InvertNeighborsListCUDA(
22 int64_t num_points,
23 const torch::Tensor& inp_neighbors_index,
24 const torch::Tensor& inp_neighbors_row_splits,
25 const torch::Tensor& inp_neighbors_attributes);
26#endif
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