SparseConvTranspose.h File Reference

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

#include <tbb/parallel_for.h>
#include <Eigen/Core>

Go to the source code of this file.

Namespaces
	open3d
	open3d::ml
	open3d::ml::impl

Macros
#define	FN_PARAMETERS
#define	CALL_TEMPLATE(NORMALIZE)
#define	CALL_TEMPLATE2

Functions
template<class TFeat , class TOut , class TIndex , class TKernelIndex , bool NORMALIZE>
void	open3d::ml::impl::_SparseConvTransposeComputeFeaturesCPU (TOut *out_features, const std::vector< int > &filter_dims, const TFeat *filter, size_t num_out, const TFeat *out_importance, size_t num_inp, const TFeat *inp_features, const TFeat *inp_neighbors_importance_sum, const int64_t *inp_neighbors_row_splits, const TIndex *neighbor_index, const TKernelIndex *neighbors_kernel_index, const TFeat *neighbor_importance, const int64_t *neighbors_row_splits)
template<class TFeat , class TOut , class TIndex , class TKernelIndex >
void	open3d::ml::impl::SparseConvTransposeComputeFeaturesCPU (TOut *out_features, const std::vector< int > &filter_dims, const TFeat *filter, size_t num_out, const TFeat *out_importance, size_t num_inp, const TFeat *inp_features, const TFeat *inp_neighbors_importance_sum, const int64_t *inp_neighbors_row_splits, const TIndex *neighbor_index, const TKernelIndex *neighbors_kernel_index, const TFeat *neighbor_importance, const int64_t *neighbors_row_splits, bool normalize)

Macro Definition Documentation

CALL_TEMPLATE

#define CALL_TEMPLATE

(

NORMALIZE

)

Value:

    if (NORMALIZE == normalize)                                          \
        _SparseConvTransposeComputeFeaturesCPU<TFeat, TOut, TIndex,      \
                                               TKernelIndex, NORMALIZE>( \
                FN_PARAMETERS);

CALL_TEMPLATE2

#define CALL_TEMPLATE2

Value:

    CALL_TEMPLATE(true) \
    CALL_TEMPLATE(false)

FN_PARAMETERS

#define FN_PARAMETERS

Value:

    out_features, filter_dims, filter, num_out, out_importance, num_inp,      \
            inp_features, inp_neighbors_importance_sum,                       \
            inp_neighbors_row_splits, neighbor_index, neighbors_kernel_index, \
            neighbor_importance, neighbors_row_splits