29#include "torch/script.h"
33 const torch::Tensor& row_splits);
35#ifdef BUILD_CUDA_MODULE
37torch::Tensor ReduceSubarraysSumCUDA(
const torch::Tensor& values,
38 const torch::Tensor& row_splits);
torch::Tensor ReduceSubarraysSumCPU(const torch::Tensor &values, const torch::Tensor &row_splits)
Definition: ReduceSubarraysSumOpKernel.cpp:35