VoxelPoolingOpKernel.cpp File Reference

(b9e049c (Sun Dec 31 11:36:26 2023 -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);

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,   \
            const AccumulationFn, const AccumulationFn);

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
	)