BuildSpatialHashTableOps.cpp File Reference

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

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

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

Functions
template<class T >
void	BuildSpatialHashTableCPU (const torch::Tensor &points, double radius, const torch::Tensor &points_row_splits, const std::vector< uint32_t > &hash_table_splits, torch::Tensor &hash_table_index, torch::Tensor &hash_table_cell_splits)
std::tuple< torch::Tensor, torch::Tensor, torch::Tensor >	BuildSpatialHashTable (torch::Tensor points, double radius, torch::Tensor points_row_splits, double hash_table_size_factor, int64_t max_hash_table_size)

Macro Definition Documentation

CALL

#define CALL	(		type,
			fn
	)

Value:

    if (CompareTorchDtype<type>(point_type)) {                           \
        fn<type>(FN_PARAMETERS);                                         \
        return std::make_tuple(hash_table_index, hash_table_cell_splits, \
                               out_hash_table_splits);                   \
    }

FN_PARAMETERS

#define FN_PARAMETERS

Value:

    points, radius, points_row_splits, hash_table_splits, hash_table_index, \
            hash_table_cell_splits

Function Documentation

BuildSpatialHashTable()

std::tuple<torch::Tensor, torch::Tensor, torch::Tensor> BuildSpatialHashTable	(	torch::Tensor	points,
		double	radius,
		torch::Tensor	points_row_splits,
		double	hash_table_size_factor,
		int64_t	max_hash_table_size
	)

BuildSpatialHashTableCPU()

template<class T >

void BuildSpatialHashTableCPU	(	const torch::Tensor &	points,
		double	radius,
		const torch::Tensor &	points_row_splits,
		const std::vector< uint32_t > &	hash_table_splits,
		torch::Tensor &	hash_table_index,
		torch::Tensor &	hash_table_cell_splits
	)