ReduceSubarraysSumOpKernel.h File Reference

(1e7b174 (Tue Jan 7 15:20:18 2025 -0800))

#include "torch/script.h"

Go to the source code of this file.

Functions
template<class T >
torch::Tensor	ReduceSubarraysSumCPU (const torch::Tensor &values, const torch::Tensor &row_splits)

Function Documentation

ReduceSubarraysSumCPU()

template<class T >

torch::Tensor ReduceSubarraysSumCPU	(	const torch::Tensor &	values,
		const torch::Tensor &	row_splits
	)