Open3D (C++ API)  0.18.0+252c867
Macros | Functions
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 
)