Function cudarc::cudnn::result::get_convolution_backward_filter_workspace_size

pub unsafe fn get_convolution_backward_filter_workspace_size(
    handle: cudnnHandle_t,
    x_desc: cudnnTensorDescriptor_t,
    dy_desc: cudnnTensorDescriptor_t,
    conv_desc: cudnnConvolutionDescriptor_t,
    grad_desc: cudnnFilterDescriptor_t,
    algo: cudnnConvolutionBwdFilterAlgo_t
) -> Result<usize, CudnnError>

Returns size in bytes. See nvidia docs

Safety

• All handles & descriptors must still be allocated.
• The pointers must point to valid memory.