(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 NORMALIZE> |
void | open3d::ml::impl::_SparseConvTransposeBackpropFilterCPU (TOut *filter_backprop, const std::vector< int > &filter_dims, 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 *neighbors_index, const TKernelIndex *neighbors_kernel_index, const TFeat *neighbors_importance, const int64_t *neighbors_row_splits, const TFeat *out_features_gradient) |
|
template<class TFeat , class TOut , class TIndex , class TKernelIndex > |
void | open3d::ml::impl::SparseConvTransposeBackpropFilterCPU (TOut *filter_backprop, const std::vector< int > &filter_dims, 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 *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 |
( |
|
NORMALIZE | ) |
|
Value: if (NORMALIZE == normalize) \
_SparseConvTransposeBackpropFilterCPU<TFeat, TOut, TIndex, \
TKernelIndex, NORMALIZE>( \
FN_PARAMETERS);
◆ CALL_TEMPLATE2
Value:
CALL_TEMPLATE(false)
#define CALL_TEMPLATE(NORMALIZE)
◆ FN_PARAMETERS
Value: filter_backprop, filter_dims, num_out, out_importance, num_inp, \
inp_features, inp_neighbors_importance_sum, \
inp_neighbors_row_splits, neighbors_index, neighbors_kernel_index, \
neighbors_importance, neighbors_row_splits, out_features_gradient