VoxelizeOpKernel.cpp File Reference

(b9e049c (Sun Dec 31 11:36:26 2023 -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 &row_splits, 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, torch::Tensor &voxel_batch_splits)

Macro Definition Documentation

CASE

#define CASE

(

NDIM

)

Value:

    case NDIM:                                                                \
        VoxelizeCPU<T, NDIM>(                                                 \
                points.size(0), points.data_ptr<T>(), row_splits.size(0) - 1, \
                row_splits.data_ptr<int64_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& row_splits,    \
            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,                           \
            torch::Tensor& voxel_batch_splits);

Function Documentation

VoxelizeCPU()

template<class T >

void VoxelizeCPU	(	const torch::Tensor &	points,
		const torch::Tensor &	row_splits,
		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,
		torch::Tensor &	voxel_batch_splits
	)