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)

const torch::Tensor &	VoxelCoords () const

const torch::Tensor &	VoxelPointIndices () const

const torch::Tensor &	VoxelPointRowSplits () const

Constructor & Destructor Documentation

◆ VoxelizeOutputAllocator()

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

inline

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

inline

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

inline

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

inline

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

inline

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

inline

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

inline

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