Open3D (C++ API)  0.12.0
Macros | Functions
VoxelPoolingOpKernel.cpp File Reference

(313315d (Wed Dec 23 23:39:47 2020 -0800))

#include "open3d/ml/impl/misc/VoxelPooling.h"
#include "open3d/ml/pytorch/TorchHelper.h"
#include "torch/script.h"

Macros

#define INSTANTIATE(TReal, TFeat)
 
#define INSTANTIATE(TReal, TFeat)
 

Functions

template<class TReal , class TFeat >
std::tuple< torch::Tensor, torch::Tensor > VoxelPoolingCPU (const torch::Tensor &positions, const torch::Tensor &features, const double voxel_size, const AccumulationFn position_fn, const AccumulationFn feature_fn, const bool debug)
 
template<class TReal , class TFeat >
void VoxelPoolingGradCPU (torch::Tensor &features_backprop, const torch::Tensor &positions, const torch::Tensor &features, const torch::Tensor &pooled_positions, const torch::Tensor &pooled_features_gradient, const double voxel_size, const AccumulationFn position_fn, const AccumulationFn feature_fn)
 

Macro Definition Documentation

◆ INSTANTIATE [1/2]

#define INSTANTIATE (   TReal,
  TFeat 
)
Value:
template std::tuple<torch::Tensor, torch::Tensor> \
VoxelPoolingCPU<TReal, TFeat>(const torch::Tensor&, const torch::Tensor&, \
const double, const AccumulationFn, \
const AccumulationFn, const bool);
AccumulationFn
Definition: VoxelPooling.h:40

◆ INSTANTIATE [2/2]

#define INSTANTIATE (   TReal,
  TFeat 
)
Value:
template void VoxelPoolingGradCPU<TReal, TFeat>( \
torch::Tensor&, const torch::Tensor&, const torch::Tensor&, \
const torch::Tensor&, const torch::Tensor&, const double, \
AccumulationFn
Definition: VoxelPooling.h:40

Function Documentation

◆ VoxelPoolingCPU()

template<class TReal , class TFeat >
std::tuple<torch::Tensor, torch::Tensor> VoxelPoolingCPU ( const torch::Tensor &  positions,
const torch::Tensor &  features,
const double  voxel_size,
const AccumulationFn  position_fn,
const AccumulationFn  feature_fn,
const bool  debug 
)

◆ VoxelPoolingGradCPU()

template<class TReal , class TFeat >
void VoxelPoolingGradCPU ( torch::Tensor &  features_backprop,
const torch::Tensor &  positions,
const torch::Tensor &  features,
const torch::Tensor &  pooled_positions,
const torch::Tensor &  pooled_features_gradient,
const double  voxel_size,
const AccumulationFn  position_fn,
const AccumulationFn  feature_fn 
)