Open3D (C++ API)  0.18.0
Loading...
Searching...
No Matches
ReduceSubarraysSumOpKernel.h
Go to the documentation of this file.
1// ----------------------------------------------------------------------------
2// - Open3D: www.open3d.org -
3// ----------------------------------------------------------------------------
4// Copyright (c) 2018-2023 www.open3d.org
5// SPDX-License-Identifier: MIT
6// ----------------------------------------------------------------------------
7//
8#pragma once
9
10#include "torch/script.h"
11
12template <class T>
13torch::Tensor ReduceSubarraysSumCPU(const torch::Tensor& values,
14 const torch::Tensor& row_splits);
15
16#ifdef BUILD_CUDA_MODULE
17template <class T>
18torch::Tensor ReduceSubarraysSumCUDA(const torch::Tensor& values,
19 const torch::Tensor& row_splits);
20#endif
torch::Tensor ReduceSubarraysSumCPU(const torch::Tensor &values, const torch::Tensor &row_splits)
Definition ReduceSubarraysSumOpKernel.cpp:16