Struct kn_cuda_sys::wrapper::descriptor::ConvolutionDescriptor

pub struct ConvolutionDescriptor(/* private fields */);

Implementations

impl ConvolutionDescriptor

pub fn new( pad_y: i32, pad_x: i32, stride_y: i32, stride_x: i32, dilation_y: i32, dilation_x: i32, dtype: cudnnDataType_t ) -> Self

pub fn workspace_size( &self, handle: &CudnnHandle, algo: cudnnConvolutionFwdAlgo_t, input: &TensorDescriptor, filter: &FilterDescriptor, output: &TensorDescriptor ) -> usize

pub fn output_shape( &self, input_desc: &TensorDescriptor, filter_desc: &FilterDescriptor ) -> [i32; 4]

pub unsafe fn inner(&self) -> cudnnConvolutionDescriptor_t

Trait Implementations

impl Debug for ConvolutionDescriptor

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Drop for ConvolutionDescriptor

fn drop(&mut self)

Executes the destructor for this type.