Trait ConvGemmConfig

pub trait ConvGemmConfig: GlobalConfig {
    // Required methods
    fn kernel_size(&self, dim: u32) -> u32;
    fn dilation(&self, dim: u32) -> u32;
    fn stride(&self, dim: u32) -> u32;
    fn padding(&self, dim: u32) -> i32;
    fn dimensionality(&self) -> Dimensionality;
    fn line_sizes(&self) -> MatmulLineSizes;
}

Convolution specific config, extends regular matmul Config

Required Methods

fn kernel_size(&self, dim: u32) -> u32

The size of the convolution kernel at dim

fn dilation(&self, dim: u32) -> u32

The dilation of the kernel at dim

fn stride(&self, dim: u32) -> u32

The stride of the kernel at dim

fn padding(&self, dim: u32) -> i32

The padding of the kernel at dim

fn dimensionality(&self) -> Dimensionality

The dimensionality of the kernel

fn line_sizes(&self) -> MatmulLineSizes

Dyn Compatibility

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety", so this trait is not object safe.

Implementors

impl<M: GlobalConfig> ConvGemmConfig for ConvolutionConfig<M>