Open3D (C++ API)  0.12.0
Macros | Functions
ReduceSubarraysSumOps.cpp File Reference

(313315d (Wed Dec 23 23:39:47 2020 -0800))

#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 
)