Struct ConvolutionProblem

pub struct ConvolutionProblem {

    pub m: usize,
    pub n: usize,
    pub k: usize,
    pub lhs_layout: MatrixLayout,
    pub rhs_layout: MatrixLayout,
    pub lhs_line_size: u8,
    pub rhs_line_size: u8,
    pub out_line_size: u8,
    pub kernel_size: (u32, u32),
    pub stride: (u32, u32),
    pub padding: (i32, i32),
    pub dilation: (u32, u32),
    pub batches: usize,
    pub height: usize,
    pub width: usize,
    pub channels: usize,
    pub out_h: usize,
    pub out_w: usize,
}

Description of a matmul problem to solve, regardless of actual data

Fields

m: usize

n: usize

k: usize

lhs_layout: MatrixLayout

rhs_layout: MatrixLayout

lhs_line_size: u8

rhs_line_size: u8

out_line_size: u8

kernel_size: (u32, u32)

stride: (u32, u32)

padding: (i32, i32)

dilation: (u32, u32)

batches: usize

height: usize

width: usize

channels: usize

out_h: usize

out_w: usize

Implementations

impl ConvolutionProblem

pub fn as_matmul_problem(&self) -> MatmulProblem

Trait Implementations

impl Clone for ConvolutionProblem

fn clone(&self) -> ConvolutionProblem

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ConvolutionProblem

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

Formats the value using the given formatter.