Open3D (C++ API)  0.18.0+252c867
Macros | Functions
VoxelizeOpKernel.cpp File Reference

(252c867 (Sun May 5 05:39:34 2024 +0000))

#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;
int points
Definition: FilePCD.cpp:54

◆ 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 
)