InvertNeighborsListOpKernel.cpp File Reference

(b9e049c (Sun Dec 31 11:36:26 2023 -0800))

#include "open3d/ml/pytorch/misc/InvertNeighborsListOpKernel.h"
#include "open3d/ml/impl/misc/InvertNeighborsList.h"
#include "open3d/ml/pytorch/TorchHelper.h"
#include "torch/script.h"

Macros
#define	INSTANTIATE(TIndex, TAttr)

Functions
template<class TIndex , class TAttr >
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)

Macro Definition Documentation

INSTANTIATE

#define INSTANTIATE	(		TIndex,
			TAttr
	)

Value:

    template std::tuple<torch::Tensor, torch::Tensor, torch::Tensor>     \
    InvertNeighborsListCPU<TIndex, TAttr>(int64_t, const torch::Tensor&, \
                                          const torch::Tensor&,          \
                                          const torch::Tensor&);

Function Documentation

InvertNeighborsListCPU()

template<class TIndex , class TAttr >

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
	)