Struct ConvolutionProblem

Source

pub struct ConvolutionProblem {Show 20 fields
    pub m: usize,
    pub n: usize,
    pub k: usize,
    pub lhs_strides: Vec<usize>,
    pub rhs_strides: Vec<usize>,
    pub lhs_layout: MatrixLayout,
    pub rhs_layout: MatrixLayout,
    pub kernel_size: Vec<u32>,
    pub stride: Vec<u32>,
    pub padding: Vec<i32>,
    pub dilation: Vec<u32>,
    pub batches: usize,
    pub channels: usize,
    pub out_channels: usize,
    pub in_shape: Vec<usize>,
    pub out_shape: Vec<usize>,
    pub padded_channels: usize,
    pub operation: ConvolutionOperation,
    pub dimensionality: Dimensionality,
    pub global_dtypes: MatmulGlobalElems,
}

Expand description

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

Fields§

§m: usize§n: usize§k: usize§lhs_strides: Vec<usize>§rhs_strides: Vec<usize>§lhs_layout: MatrixLayout§rhs_layout: MatrixLayout§kernel_size: Vec<u32>§stride: Vec<u32>§padding: Vec<i32>§dilation: Vec<u32>§batches: usize§channels: usize§out_channels: usize§in_shape: Vec<usize>§out_shape: Vec<usize>§padded_channels: usize

Channels after applying loader-specific padding

§operation: ConvolutionOperation§dimensionality: Dimensionality§global_dtypes: MatmulGlobalElems

Struct ConvolutionProblem Copy item path

Fields§

Implementations§

impl ConvolutionProblem

pub fn as_matmul_problem(&self) -> MatmulProblem

pub fn should_check_channel(&self) -> bool

pub fn should_check_spatial_bounds(&self) -> bool

Trait Implementations§

impl Clone for ConvolutionProblem

fn clone(&self) -> ConvolutionProblem

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

impl Debug for ConvolutionProblem

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

Auto Trait Implementations§

impl Freeze for ConvolutionProblem

impl RefUnwindSafe for ConvolutionProblem

impl Send for ConvolutionProblem

impl Sync for ConvolutionProblem

impl Unpin for ConvolutionProblem

impl UnwindSafe for ConvolutionProblem

Blanket Implementations§

impl<T> Any for Twhere T: 'static + ?Sized,

fn type_id(&self) -> TypeId

impl<T> Borrow<T> for Twhere T: ?Sized,

fn borrow(&self) -> &T

impl<T> BorrowMut<T> for Twhere T: ?Sized,

fn borrow_mut(&mut self) -> &mut T

impl<T> CloneToUninit for Twhere T: Clone,

unsafe fn clone_to_uninit(&self, dest: *mut u8)

impl<T> From<T> for T

fn from(t: T) -> T

impl<T, U> Into<U> for Twhere U: From<T>,

fn into(self) -> U

impl<T> IntoComptime for T

fn comptime(self) -> Self

impl<T> ToOwned for Twhere T: Clone,

type Owned = T

fn to_owned(&self) -> T

fn clone_into(&self, target: &mut T)

impl<T, U> TryFrom<U> for Twhere U: Into<T>,

type Error = Infallible

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

impl<T, U> TryInto<U> for Twhere U: TryFrom<T>,

type Error = <U as TryFrom<T>>::Error

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

impl<V, T> VZip<V> for Twhere V: MultiLane<T>,

fn vzip(self) -> V

Struct ConvolutionProblem

impl<T> Any for T
where T: 'static + ?Sized,

impl<T> Borrow<T> for T
where T: ?Sized,

impl<T> BorrowMut<T> for T
where T: ?Sized,

impl<T> CloneToUninit for T
where T: Clone,

impl<T, U> Into<U> for T
where U: From<T>,

impl<T> ToOwned for T
where T: Clone,

impl<T, U> TryFrom<U> for T
where U: Into<T>,

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

impl<V, T> VZip<V> for T
where V: MultiLane<T>,