RadiusSearchOps.cpp File Reference

(313315d (Wed Dec 23 23:39:47 2020 -0800))

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

Macros
#define	FN_PARAMETERS
#define	CALL(type, fn)

Functions
template<class T >
void	RadiusSearchCPU (const torch::Tensor &points, const torch::Tensor &queries, const torch::Tensor &radii, const torch::Tensor &points_row_splits, const torch::Tensor &queries_row_splits, const Metric metric, const bool ignore_query_point, const bool return_distances, const bool normalize_distances, torch::Tensor &neighbors_index, torch::Tensor &neighbors_row_splits, torch::Tensor &neighbors_distance)
std::tuple< torch::Tensor, torch::Tensor, torch::Tensor >	MultiRadiusSearch (torch::Tensor points, torch::Tensor queries, torch::Tensor radii, torch::Tensor points_row_splits, torch::Tensor queries_row_splits, const std::string &metric_str, const bool ignore_query_point, const bool return_distances, const bool normalize_distances)

Macro Definition Documentation

CALL

#define CALL	(		type,
			fn
	)

Value:

if (CompareTorchDtype<type>(point_type)) {                        \
        fn<type>(FN_PARAMETERS);                                      \
        return std::make_tuple(neighbors_index, neighbors_row_splits, \
                               neighbors_distance);                   \
    }

FN_PARAMETERS

#define FN_PARAMETERS

Value:

points, queries, radii, points_row_splits, queries_row_splits, metric, \
            ignore_query_point, return_distances, normalize_distances,     \
            neighbors_index, neighbors_row_splits, neighbors_distance

Function Documentation

MultiRadiusSearch()

std::tuple<torch::Tensor, torch::Tensor, torch::Tensor> MultiRadiusSearch	(	torch::Tensor	points,
		torch::Tensor	queries,
		torch::Tensor	radii,
		torch::Tensor	points_row_splits,
		torch::Tensor	queries_row_splits,
		const std::string &	metric_str,
		const bool	ignore_query_point,
		const bool	return_distances,
		const bool	normalize_distances
	)

RadiusSearchCPU()

template<class T >

void RadiusSearchCPU	(	const torch::Tensor &	points,
		const torch::Tensor &	queries,
		const torch::Tensor &	radii,
		const torch::Tensor &	points_row_splits,
		const torch::Tensor &	queries_row_splits,
		const Metric	metric,
		const bool	ignore_query_point,
		const bool	return_distances,
		const bool	normalize_distances,
		torch::Tensor &	neighbors_index,
		torch::Tensor &	neighbors_row_splits,
		torch::Tensor &	neighbors_distance
	)