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;
}

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

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 HomogeneousConfig<M>