Struct ConvolutionConfig 

pub struct ConvolutionConfig<M: GlobalConfig> {
    pub matmul: M,
    pub params: ConvolutionParams,
}

Fields

matmul: M

params: ConvolutionParams

Implementations

impl<M: GlobalConfig> ConvolutionConfig<M>

pub fn new( matmul: M, kernel_size: &[u32], stride: &[u32], dilation: &[u32], padding: &[i32], dim: Dimensionality, operation: ConvolutionOperation, ) -> 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 vector_sizes(&self) -> MatmulVectorSizes

fn cube_dim(&self) -> CubeDim

fn check_spatial_bounds(&self) -> bool

fn params(&self) -> ConvolutionParams

The size of the convolution kernel at dim

fn operation(&self) -> ConvolutionOperation

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>