|
| ReduceSubarraysSumOpKernelCPU (OpKernelConstruction *construction) |
|
void | Kernel (OpKernelContext *context, const tensorflow::Tensor &values, const tensorflow::Tensor &row_splits, tensorflow::Tensor &sums) |
|
◆ ReduceSubarraysSumOpKernelCPU()
◆ Kernel()
template<class T >
void ReduceSubarraysSumOpKernelCPU< T >::Kernel |
( |
OpKernelContext * |
context, |
|
|
const tensorflow::Tensor & |
values, |
|
|
const tensorflow::Tensor & |
row_splits, |
|
|
tensorflow::Tensor & |
sums |
|
) |
| |
|
inline |
The documentation for this class was generated from the following file: