ContinuousConvTransposeOpKernel.h File Reference

(252c867 (Sun May 5 05:39:34 2024 +0000))

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

Go to the source code of this file.

Functions
template<class TFeat , class TOut , class TReal , class TIndex >
void	ContinuousConvTransposeCPU (const torch::Tensor &filters, const torch::Tensor &out_positions, const torch::Tensor &out_importance, const torch::Tensor &extents, const torch::Tensor &offset, const torch::Tensor &inp_positions, const torch::Tensor &inp_features, const torch::Tensor &inp_neighbors_index, const torch::Tensor &inp_neighbors_importance_sum, const torch::Tensor &inp_neighbors_row_splits, const torch::Tensor &neighbors_index, const torch::Tensor &neighbors_importance, const torch::Tensor &neighbors_row_splits, 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 &out_features)

Function Documentation

ContinuousConvTransposeCPU()

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

void ContinuousConvTransposeCPU	(	const torch::Tensor &	filters,
		const torch::Tensor &	out_positions,
		const torch::Tensor &	out_importance,
		const torch::Tensor &	extents,
		const torch::Tensor &	offset,
		const torch::Tensor &	inp_positions,
		const torch::Tensor &	inp_features,
		const torch::Tensor &	inp_neighbors_index,
		const torch::Tensor &	inp_neighbors_importance_sum,
		const torch::Tensor &	inp_neighbors_row_splits,
		const torch::Tensor &	neighbors_index,
		const torch::Tensor &	neighbors_importance,
		const torch::Tensor &	neighbors_row_splits,
		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 &	out_features
	)