InvertNeighborsListOps.cpp File Reference

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

#include "open3d/ml/pytorch/misc/InvertNeighborsListOps.h"
#include <vector>
#include "open3d/ml/pytorch/TorchHelper.h"
#include "open3d/ml/pytorch/misc/InvertNeighborsListOpKernel.h"
#include "torch/script.h"

Macros
#define	FN_PARAMETERS
#define	CALL(idx_t, attr_t, fn)

Functions
std::tuple< torch::Tensor, torch::Tensor, torch::Tensor >	InvertNeighborsList (int64_t num_points, torch::Tensor inp_neighbors_index, torch::Tensor inp_neighbors_row_splits, torch::Tensor inp_neighbors_attributes)

Macro Definition Documentation

CALL

#define CALL	(		idx_t,
			attr_t,
			fn
	)

Value:

    if (CompareTorchDtype<idx_t>(index_type) &&  \
        CompareTorchDtype<attr_t>(attr_type)) {  \
        return fn<idx_t, attr_t>(FN_PARAMETERS); \
    }

FN_PARAMETERS

#define FN_PARAMETERS

Value:

    num_points, inp_neighbors_index, inp_neighbors_row_splits, \
            inp_neighbors_attributes

Function Documentation

InvertNeighborsList()

std::tuple<torch::Tensor, torch::Tensor, torch::Tensor> InvertNeighborsList	(	int64_t	num_points,
		torch::Tensor	inp_neighbors_index,
		torch::Tensor	inp_neighbors_row_splits,
		torch::Tensor	inp_neighbors_attributes
	)