ContinuousConvBackpropFilterOpKernel.cpp File Reference

#include <vector>
#include "open3d/ml/impl/continuous_conv/ContinuousConvBackpropFilter.h"
#include "torch/script.h"

Macros
#define	INSTANTIATE(TFeat, TOut, TReal, TIndex)

Functions
template<class TFeat, class TOut, class TReal, class TIndex>
void	ContinuousConvBackpropFilterCPU (const torch::Tensor &filters, const torch::Tensor &out_positions, const torch::Tensor &extents, const torch::Tensor &offset, const torch::Tensor &inp_positions, const torch::Tensor &inp_features, const torch::Tensor &inp_importance, const torch::Tensor &neighbors_index, const torch::Tensor &neighbors_importance, const torch::Tensor &neighbors_row_splits, const torch::Tensor &out_features_gradient, const bool align_corners, const open3d::ml::impl::CoordinateMapping coordinate_mapping, const bool normalize, const open3d::ml::impl::InterpolationMode interpolation, const int64_t max_temp_mem_MB, torch::Tensor &filter_backprop)

Macro Definition Documentation

INSTANTIATE

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

Value:

    template void ContinuousConvBackpropFilterCPU<TFeat, TOut, TReal, TIndex>( \
            const torch::Tensor& filters, const torch::Tensor& out_positions,  \
            const torch::Tensor& extents, const torch::Tensor& offset,         \
            const torch::Tensor& inp_positions,                                \
            const torch::Tensor& inp_features,                                 \
            const torch::Tensor& inp_importance,                               \
            const torch::Tensor& neighbors_index,                              \
            const torch::Tensor& neighbors_importance,                         \
            const torch::Tensor& neighbors_row_splits,                         \
            const torch::Tensor& out_features_gradient,                        \
            const bool align_corners,                                          \
            const open3d::ml::impl::CoordinateMapping coordinate_mapping,      \
            const bool normalize,                                              \
            const open3d::ml::impl::InterpolationMode interpolation,           \
            const int64_t max_temp_mem_MB, torch::Tensor& filter_backprop);

Function Documentation

ContinuousConvBackpropFilterCPU()

template<class TFeat, class TOut, class TReal, class TIndex>

void ContinuousConvBackpropFilterCPU	(	const torch::Tensor &	filters,
		const torch::Tensor &	out_positions,
		const torch::Tensor &	extents,
		const torch::Tensor &	offset,
		const torch::Tensor &	inp_positions,
		const torch::Tensor &	inp_features,
		const torch::Tensor &	inp_importance,
		const torch::Tensor &	neighbors_index,
		const torch::Tensor &	neighbors_importance,
		const torch::Tensor &	neighbors_row_splits,
		const torch::Tensor &	out_features_gradient,
		const bool	align_corners,
		const open3d::ml::impl::CoordinateMapping	coordinate_mapping,
		const bool	normalize,
		const open3d::ml::impl::InterpolationMode	interpolation,
		const int64_t	max_temp_mem_MB,
		torch::Tensor &	filter_backprop )