Function cudarc::cudnn::result::get_reduction_workspace_size

pub unsafe fn get_reduction_workspace_size(
    handle: cudnnHandle_t,
    reduce_tensor_desc: cudnnReduceTensorDescriptor_t,
    a_desc: cudnnTensorDescriptor_t,
    c_desc: cudnnTensorDescriptor_t
) -> Result<usize, CudnnError>

See nvidia docs

Safety

Handle and descriptors must be properly allocated and not freed already.