ReduceSubarraysSumOps.cpp File Reference

(252c867 (Sun May 5 05:39:34 2024 +0000))

#include "open3d/ml/pytorch/misc/ReduceSubarraysSumOps.h"
#include <vector>
#include "open3d/ml/pytorch/TorchHelper.h"
#include "open3d/ml/pytorch/misc/ReduceSubarraysSumOpKernel.h"
#include "torch/script.h"

Macros
#define	CALL(attr_t, fn)

Functions
torch::Tensor	ReduceSubarraysSum (torch::Tensor values, torch::Tensor row_splits)

Macro Definition Documentation

CALL

#define CALL	(		attr_t,
			fn
	)

Value:

    if (CompareTorchDtype<attr_t>(attr_type)) { \
        return fn<attr_t>(values, row_splits);  \
    }

Function Documentation

ReduceSubarraysSum()

torch::Tensor ReduceSubarraysSum	(	torch::Tensor	values,
		torch::Tensor	row_splits
	)