VoxelizeOutputAllocator Class Reference

#include <VoxelizeOpKernel.h>

Public Member Functions
	VoxelizeOutputAllocator (torch::DeviceType device_type, int device_idx)
void	AllocVoxelCoords (int32_t **ptr, int64_t rows, int64_t cols)
void	AllocVoxelPointIndices (int64_t **ptr, int64_t num)
void	AllocVoxelPointRowSplits (int64_t **ptr, int64_t num)
void	AllocVoxelBatchSplits (int64_t **ptr, int64_t num)
const torch::Tensor &	VoxelCoords () const
const torch::Tensor &	VoxelPointIndices () const
const torch::Tensor &	VoxelPointRowSplits () const
const torch::Tensor &	VoxelBatchSplits () const

Constructor & Destructor Documentation

VoxelizeOutputAllocator()

VoxelizeOutputAllocator::VoxelizeOutputAllocator ( torch::DeviceType device_type, int device_idx )

inline

Member Function Documentation

AllocVoxelBatchSplits()

void VoxelizeOutputAllocator::AllocVoxelBatchSplits ( int64_t ** ptr, int64_t num )

inline

AllocVoxelCoords()

void VoxelizeOutputAllocator::AllocVoxelCoords ( int32_t ** ptr, int64_t rows, int64_t cols )

inline

AllocVoxelPointIndices()

void VoxelizeOutputAllocator::AllocVoxelPointIndices ( int64_t ** ptr, int64_t num )

inline

AllocVoxelPointRowSplits()

void VoxelizeOutputAllocator::AllocVoxelPointRowSplits ( int64_t ** ptr, int64_t num )

inline

VoxelBatchSplits()

const torch::Tensor & VoxelizeOutputAllocator::VoxelBatchSplits ( ) const

inline

VoxelCoords()

const torch::Tensor & VoxelizeOutputAllocator::VoxelCoords ( ) const

inline

VoxelPointIndices()

const torch::Tensor & VoxelizeOutputAllocator::VoxelPointIndices ( ) const

inline

VoxelPointRowSplits()

const torch::Tensor & VoxelizeOutputAllocator::VoxelPointRowSplits ( ) const

inline

---

The documentation for this class was generated from the following file:

• cpp/open3d/ml/pytorch/misc/VoxelizeOpKernel.h