SparseConvBackpropFilter.h File Reference

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

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

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

Macro Definition Documentation

CALL_TEMPLATE

#define CALL_TEMPLATE

(

HAS_IMPORTANCE

)

Value:

    if (HAS_IMPORTANCE == has_importance)                              \
        _SparseConvBackropFilterCPU<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:

    filter_backprop, filter_dims, num_out, num_inp, inp_features,              \
            inp_importance, neighbors_index, neighbors_kernel_index,           \
            neighbors_importance, neighbors_row_splits, out_features_gradient, \
            normalize