#include <VoxelizeOpKernel.h>
◆ VoxelizeOutputAllocator()
VoxelizeOutputAllocator::VoxelizeOutputAllocator |
( |
torch::DeviceType |
device_type, |
|
|
int |
device_idx |
|
) |
| |
|
inline |
◆ 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: