SparseConvTransposeBackpropFilterOpKernel.cpp File Reference

(b9e049c (Sun Dec 31 11:36:26 2023 -0800))

#include <torch/script.h>
#include <vector>
#include "open3d/ml/impl/sparse_conv/SparseConvTransposeBackpropFilter.h"

Macros
#define	INSTANTIATE(TFeat, TOut, TIndex, TKernelIndex)

Functions
template<class TFeat , class TOut , class TIndex , class TKernelIndex >
void	SparseConvTransposeBackpropFilterCPU (const torch::Tensor &filters, const torch::Tensor &out_importance, const torch::Tensor &inp_features, const torch::Tensor &inp_neighbors_importance_sum, const torch::Tensor &inp_neighbors_row_splits, const torch::Tensor &neighbors_index, const torch::Tensor &neighbors_kernel_index, const torch::Tensor &neighbors_importance, const torch::Tensor &neighbors_row_splits, const torch::Tensor &out_features_gradient, const bool normalize, const int64_t max_temp_mem_MB, torch::Tensor &filter_backprop)

Macro Definition Documentation

INSTANTIATE

#define INSTANTIATE	(		TFeat,
			TOut,
			TIndex,
			TKernelIndex
	)

Value:

    template void                                                              \
    SparseConvTransposeBackpropFilterCPU<TFeat, TOut, TIndex, TKernelIndex>(   \
            const torch::Tensor& filters, const torch::Tensor& out_importance, \
            const torch::Tensor& inp_features,                                 \
            const torch::Tensor& inp_neighbors_importance_sum,                 \
            const torch::Tensor& inp_neighbors_row_splits,                     \
            const torch::Tensor& neighbors_index,                              \
            const torch::Tensor& neighbors_kernel_index,                       \
            const torch::Tensor& neighbors_importance,                         \
            const torch::Tensor& neighbors_row_splits,                         \
            const torch::Tensor& out_features_gradient, const bool normalize,  \
            const int64_t max_temp_mem_MB, torch::Tensor& filter_backprop);

Function Documentation

SparseConvTransposeBackpropFilterCPU()

template<class TFeat , class TOut , class TIndex , class TKernelIndex >

void SparseConvTransposeBackpropFilterCPU	(	const torch::Tensor &	filters,
		const torch::Tensor &	out_importance,
		const torch::Tensor &	inp_features,
		const torch::Tensor &	inp_neighbors_importance_sum,
		const torch::Tensor &	inp_neighbors_row_splits,
		const torch::Tensor &	neighbors_index,
		const torch::Tensor &	neighbors_kernel_index,
		const torch::Tensor &	neighbors_importance,
		const torch::Tensor &	neighbors_row_splits,
		const torch::Tensor &	out_features_gradient,
		const bool	normalize,
		const int64_t	max_temp_mem_MB,
		torch::Tensor &	filter_backprop
	)