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

#include "open3d/ml/pytorch/misc/VoxelizeOpKernel.h"
#include "open3d/ml/impl/misc/Voxelize.h"
#include "open3d/ml/pytorch/TorchHelper.h"
#include "torch/script.h"

Macros
#define	CASE(NDIM)

#define	INSTANTIATE(T)

Functions
template<class T >
void	VoxelizeCPU (const torch::Tensor &points, const torch::Tensor &voxel_size, const torch::Tensor &points_range_min, const torch::Tensor &points_range_max, const int64_t max_points_per_voxel, const int64_t max_voxels, torch::Tensor &voxel_coords, torch::Tensor &voxel_point_indices, torch::Tensor &voxel_point_row_splits)

Macro Definition Documentation

◆ CASE

#define CASE ( NDIM )

Value:

case NDIM:                                                            \
        VoxelizeCPU<T, NDIM>(                                             \
                points.size(0), points.data_ptr<T>(),                     \
                voxel_size.data_ptr<T>(), points_range_min.data_ptr<T>(), \
                points_range_max.data_ptr<T>(), max_points_per_voxel,     \
                max_voxels, output_allocator);                            \
        break;

◆ INSTANTIATE

#define INSTANTIATE ( T )

Value:

template void VoxelizeCPU<T>(                                            \
            const torch::Tensor& points, const torch::Tensor& voxel_size,    \
            const torch::Tensor& points_range_min,                           \
            const torch::Tensor& points_range_max,                           \
            const int64_t max_points_per_voxel, const int64_t max_voxels,    \
            torch::Tensor& voxel_coords, torch::Tensor& voxel_point_indices, \
            torch::Tensor& voxel_point_row_splits);

Function Documentation

◆ VoxelizeCPU()

template<class T >

void VoxelizeCPU	(	const torch::Tensor &	points,
		const torch::Tensor &	voxel_size,
		const torch::Tensor &	points_range_min,
		const torch::Tensor &	points_range_max,
		const int64_t	max_points_per_voxel,
		const int64_t	max_voxels,
		torch::Tensor &	voxel_coords,
		torch::Tensor &	voxel_point_indices,
		torch::Tensor &	voxel_point_row_splits
	)