ReduceSubarraysSumOpKernel.cpp File Reference

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

#include "open3d/ml/pytorch/misc/ReduceSubarraysSumOpKernel.h"
#include "open3d/ml/impl/misc/ReduceSubarraysSum.h"
#include "open3d/ml/pytorch/TorchHelper.h"
#include "torch/script.h"

Macros
#define	INSTANTIATE(T)

Functions
template<class T >
torch::Tensor	ReduceSubarraysSumCPU (const torch::Tensor &values, const torch::Tensor &row_splits)

Macro Definition Documentation

INSTANTIATE

#define INSTANTIATE

(

T

)

Value:

    template torch::Tensor ReduceSubarraysSumCPU<T>(const torch::Tensor&, \
                                                    const torch::Tensor&);

Function Documentation

ReduceSubarraysSumCPU()

template<class T >

torch::Tensor ReduceSubarraysSumCPU	(	const torch::Tensor &	values,
		const torch::Tensor &	row_splits
	)