VoxelizeOpKernel.h File Reference

(9238339 (Tue Mar 14 18:49:09 2023 -0700))

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

Go to the source code of this file.

Data Structures
class	VoxelizeOutputAllocator

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)

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
	)