Open3D (C++ API)  0.18.0+252c867
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 
12 template <class T>
13 torch::Tensor ReduceSubarraysSumCPU(const torch::Tensor& values,
14  const torch::Tensor& row_splits);
15 
16 #ifdef BUILD_CUDA_MODULE
17 template <class T>
18 torch::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