FixedRadiusSearchOpKernel.cpp File Reference

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

#include "open3d/core/nns/FixedRadiusSearchImpl.h"
#include "open3d/core/nns/NeighborSearchCommon.h"
#include "open3d/ml/pytorch/TorchHelper.h"
#include "open3d/ml/pytorch/misc/NeighborSearchAllocator.h"
#include "torch/script.h"

Macros
#define	INSTANTIATE(T, TIndex)

Functions
template<class T , class TIndex >
void	FixedRadiusSearchCPU (const torch::Tensor &points, const torch::Tensor &queries, double radius, const torch::Tensor &points_row_splits, const torch::Tensor &queries_row_splits, const torch::Tensor &hash_table_splits, const torch::Tensor &hash_table_index, const torch::Tensor &hash_table_cell_splits, const Metric metric, const bool ignore_query_point, const bool return_distances, torch::Tensor &neighbors_index, torch::Tensor &neighbors_row_splits, torch::Tensor &neighbors_distance)

Macro Definition Documentation

INSTANTIATE

#define INSTANTIATE	(		T,
			TIndex
	)

Value:

    template void FixedRadiusSearchCPU<T, TIndex>(                            \
            const torch::Tensor& points, const torch::Tensor& queries,        \
            double radius, const torch::Tensor& points_row_splits,            \
            const torch::Tensor& queries_row_splits,                          \
            const torch::Tensor& hash_table_splits,                           \
            const torch::Tensor& hash_table_index,                            \
            const torch::Tensor& hash_table_cell_splits, const Metric metric, \
            const bool ignore_query_point, const bool return_distances,       \
            torch::Tensor& neighbors_index,                                   \
            torch::Tensor& neighbors_row_splits,                              \
            torch::Tensor& neighbors_distance);

Function Documentation

FixedRadiusSearchCPU()

template<class T , class TIndex >

void FixedRadiusSearchCPU	(	const torch::Tensor &	points,
		const torch::Tensor &	queries,
		double	radius,
		const torch::Tensor &	points_row_splits,
		const torch::Tensor &	queries_row_splits,
		const torch::Tensor &	hash_table_splits,
		const torch::Tensor &	hash_table_index,
		const torch::Tensor &	hash_table_cell_splits,
		const Metric	metric,
		const bool	ignore_query_point,
		const bool	return_distances,
		torch::Tensor &	neighbors_index,
		torch::Tensor &	neighbors_row_splits,
		torch::Tensor &	neighbors_distance
	)