#include "open3d/ml/pytorch/TorchHelper.h"
#include "torch/script.h"
Go to the source code of this file.
|
| 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) |
◆ 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 ) |