BuildSpatialHashTableOpKernel.cpp File Reference

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

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

Macros
#define	INSTANTIATE(T)

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)

Macro Definition Documentation

INSTANTIATE

#define INSTANTIATE

(

T

)

Value:

    template void BuildSpatialHashTableCPU<T>(                  \
            const torch::Tensor&, double, const torch::Tensor&, \
            const std::vector<uint32_t>&, torch::Tensor&, torch::Tensor&);

Function Documentation

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
	)