SparseConvTranspose.h File Reference

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

Go to the source code of this file.

Namespaces
namespace	open3d
namespace	open3d::ml
namespace	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