SparseConvTransposeOps.cpp File Reference

(b9e049c (Sun Dec 31 11:36:26 2023 -0800))

#include <torch/script.h>
#include <vector>
#include "open3d/ml/pytorch/TorchHelper.h"
#include "open3d/ml/pytorch/misc/InvertNeighborsListOps.h"
#include "open3d/ml/pytorch/misc/ReduceSubarraysSumOps.h"
#include "open3d/ml/pytorch/sparse_conv/SparseConvOpKernel.h"
#include "open3d/ml/pytorch/sparse_conv/SparseConvTransposeBackpropFilterOpKernel.h"
#include "open3d/ml/pytorch/sparse_conv/SparseConvTransposeOpKernel.h"

Data Structures
class	SparseConvTransposeFunction

Macros
#define	FN_PARAMETERS
#define	CALL(feat_t, out_t, index_t, kernel_index_t, fn)
#define	CALL(feat_t, out_t, index_t, kernel_index_t, fn_suffix)

Functions
torch::Tensor	SparseConvTranspose (const torch::Tensor &filters, const torch::Tensor &out_importance, 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_kernel_index, const torch::Tensor &neighbors_importance, const torch::Tensor &neighbors_row_splits, const bool normalize, const int64_t max_temp_mem_MB)

Macro Definition Documentation

CALL [1/2]

#define CALL	(		feat_t,
			out_t,
			index_t,
			kernel_index_t,
			fn
	)

Value:

    if (CompareTorchDtype<feat_t>(feat_dtype) &&                   \
        CompareTorchDtype<index_t>(index_dtype) &&                 \
        CompareTorchDtype<kernel_index_t>(kernel_index_dtype)) {   \
        fn<feat_t, out_t, index_t, kernel_index_t>(FN_PARAMETERS); \
        return out_features;                                       \
    }

CALL [2/2]

#define CALL	(		feat_t,
			out_t,
			index_t,
			kernel_index_t,
			fn_suffix
	)

FN_PARAMETERS

#define FN_PARAMETERS

Value:

    filters, out_importance, inp_features, inp_neighbors_importance_sum,       \
            inp_neighbors_row_splits, neighbors_index, neighbors_kernel_index, \
            neighbors_importance, neighbors_row_splits, normalize,             \
            max_temp_mem_MB, out_features

Function Documentation

SparseConvTranspose()

torch::Tensor SparseConvTranspose	(	const torch::Tensor &	filters,
		const torch::Tensor &	out_importance,
		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_kernel_index,
		const torch::Tensor &	neighbors_importance,
		const torch::Tensor &	neighbors_row_splits,
		const bool	normalize,
		const int64_t	max_temp_mem_MB
	)