SparseConvOpKernel.h File Reference

#include <torch/script.h>
#include <vector>

Go to the source code of this file.

Functions
template<class TFeat, class TOut, class TIndex, class TKernelIndex>
void	SparseConvCPU (const torch::Tensor &filters, const torch::Tensor &inp_features, const torch::Tensor &inp_importance, const torch::Tensor &neighbors_index, const torch::Tensor &neighbors_kernel_index, const torch::Tensor &neighbors_importance, const torch::Tensor &neighbors_row_splits, const bool normalize, const int64_t max_temp_mem_MB, torch::Tensor &out_features)

Function Documentation

SparseConvCPU()

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

void SparseConvCPU	(	const torch::Tensor &	filters,
		const torch::Tensor &	inp_features,
		const torch::Tensor &	inp_importance,
		const torch::Tensor &	neighbors_index,
		const torch::Tensor &	neighbors_kernel_index,
		const torch::Tensor &	neighbors_importance,
		const torch::Tensor &	neighbors_row_splits,
		const bool	normalize,
		const int64_t	max_temp_mem_MB,
		torch::Tensor &	out_features )