(b9e049c (Sun Dec 31 11:36:26 2023 -0800))
#include <tbb/parallel_for.h>
#include <Eigen/Core>
#include <mutex>
Go to the source code of this file.
|
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) |
|
◆ CALL_TEMPLATE
#define CALL_TEMPLATE |
( |
|
HAS_IMPORTANCE | ) |
|
Value: if (HAS_IMPORTANCE == has_importance) \
_SparseConvBackropFilterCPU<TFeat, TOut, TIndex, TKernelIndex, \
◆ CALL_TEMPLATE2
Value:
CALL_TEMPLATE(false)
#define CALL_TEMPLATE(HAS_IMPORTANCE)
◆ 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