ContinuousConvTransposeOps.cpp File Reference

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

#include <vector>
#include "open3d/ml/pytorch/TorchHelper.h"
#include "open3d/ml/pytorch/continuous_conv/ContinuousConvHelper.h"
#include "open3d/ml/pytorch/continuous_conv/ContinuousConvOpKernel.h"
#include "open3d/ml/pytorch/continuous_conv/ContinuousConvTransposeBackpropFilterOpKernel.h"
#include "open3d/ml/pytorch/continuous_conv/ContinuousConvTransposeOpKernel.h"
#include "open3d/ml/pytorch/misc/InvertNeighborsListOps.h"
#include "open3d/ml/pytorch/misc/ReduceSubarraysSumOps.h"
#include "torch/script.h"

Data Structures
class	ContinuousConvTransposeFunction

Macros
#define	FN_PARAMETERS
#define	CALL(real_t, index_t, fn)
#define	CALL(real_t, index_t, fn_suffix)

Functions
torch::Tensor	ContinuousConvTranspose (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 std::string &coordinate_mapping_str, const bool normalize, const std::string &interpolation_str, const int64_t max_temp_mem_MB)

Macro Definition Documentation

CALL [1/2]

#define CALL	(		real_t,
			index_t,
			fn
	)

Value:

if (CompareTorchDtype<real_t>(real_dtype) &&   \
        CompareTorchDtype<index_t>(index_dtype)) { \
        fn<real_t, index_t>(FN_PARAMETERS);        \
        return out_features;                       \
    }

CALL [2/2]

#define CALL	(		real_t,
			index_t,
			fn_suffix
	)

FN_PARAMETERS

#define FN_PARAMETERS

Value:

filters, out_positions, out_importance, extents, offset, inp_positions,  \
            inp_features, inp_neighbors_index, inp_neighbors_importance_sum, \
            inp_neighbors_row_splits, neighbors_index, neighbors_importance, \
            neighbors_row_splits, align_corners, coordinate_mapping,         \
            normalize, interpolation, max_temp_mem_MB, out_features

Function Documentation

ContinuousConvTranspose()

torch::Tensor ContinuousConvTranspose	(	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 std::string &	coordinate_mapping_str,
		const bool	normalize,
		const std::string &	interpolation_str,
		const int64_t	max_temp_mem_MB
	)