SparseConv.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(HAS_IMPORTANCE)
#define	CALL_TEMPLATE2

Functions
template<class TFeat , class TOut , class TIndex , class TKernelIndex , bool POINT_IMPORTANCE>
void	open3d::ml::impl::_SparseConvComputeFeaturesCPU (TOut *out_features, const std::vector< int > &filter_dims, const TFeat *filter, size_t num_out, size_t num_inp, const TFeat *inp_features, const TFeat *inp_importance, size_t neighbors_index_size, const TIndex *neighbors_index, const TKernelIndex *neighbors_kernel_index, const TFeat *neighbors_importance, const int64_t *neighbors_row_splits, bool normalize)
template<class TFeat , class TOut , class TIndex , class TKernelIndex >
void	open3d::ml::impl::SparseConvComputeFeaturesCPU (TOut *out_features, const std::vector< int > &filter_dims, const TFeat *filter, size_t num_out, size_t num_inp, const TFeat *inp_features, const TFeat *inp_importance, size_t neighbors_index_size, const TIndex *neighbors_index, const TKernelIndex *neighbors_kernel_index, const TFeat *neighbors_importance, const int64_t *neighbors_row_splits, bool normalize)

Macro Definition Documentation

CALL_TEMPLATE

#define CALL_TEMPLATE

(

HAS_IMPORTANCE

)

Value:

    if (HAS_IMPORTANCE == has_importance)                                \
        _SparseConvComputeFeaturesCPU<TFeat, TOut, TIndex, TKernelIndex, \
                                      HAS_IMPORTANCE>(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, num_inp, inp_features, \
            inp_importance, neighbors_index_size, neighbors_index,     \
            neighbors_kernel_index, neighbors_importance,              \
            neighbors_row_splits, normalize