Struct ConvolutionConfig 

pub struct ConvolutionConfig<M: GlobalConfig> {
    pub matmul: M,
    pub convolution_params: ConvolutionParams,
    pub num_stages: u32,
}

Fields

matmul: M

convolution_params: ConvolutionParams

num_stages: u32

Implementations

impl<M: GlobalConfig> ConvolutionConfig<M>

pub fn new( matmul: M, kernel_size: &[u32], stride: &[u32], dilation: &[u32], padding: &[i32], dim: Dimensionality, num_stages: u32, ) -> Result<Self, MatmulSetupError>

Trait Implementations

impl<M: Clone + GlobalConfig> Clone for ConvolutionConfig<M>

fn clone(&self) -> ConvolutionConfig<M>

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

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

type GlobalMatmulConfig = M

fn matmul_config(&self) -> Self::GlobalMatmulConfig

fn line_sizes(&self) -> MatmulLineSizes

fn cube_dim(&self) -> CubeDim

fn check_spatial_bounds(&self) -> bool

fn convolution_params(&self) -> ConvolutionParams

The size of the convolution kernel at dim

fn lhs_global_memory_config(&self) -> GlobalMemoryConfig

fn rhs_global_memory_config(&self) -> GlobalMemoryConfig

fn out_global_memory_config(&self) -> GlobalMemoryConfig

impl<M: Debug + GlobalConfig> Debug for ConvolutionConfig<M>

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

Formats the value using the given formatter.

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

type Target = M

The resulting type after dereferencing.

fn deref(&self) -> &Self::Target

Dereferences the value.

impl<M: Hash + GlobalConfig> Hash for ConvolutionConfig<M>

fn hash<__H: Hasher>(&self, state: &mut __H)

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl<M: PartialEq + GlobalConfig> PartialEq for ConvolutionConfig<M>

fn eq(&self, other: &ConvolutionConfig<M>) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl<M: Copy + GlobalConfig> Copy for ConvolutionConfig<M>

impl<M: Eq + GlobalConfig> Eq for ConvolutionConfig<M>

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