Skip to main content

Conv2d

cortiq_engine::vae

Struct Conv2d

pub struct Conv2d {
    pub w: Vec<f32>,
    pub b: Vec<f32>,
    pub oc: usize,
    pub ic: usize,
    pub k: usize,
}

Expand description

2-D convolution, stride 1, square kernel, symmetric padding (pad = k/2). Parallel over output channels.

Fields§

§w: Vec<f32>§b: Vec<f32>§oc: usize§ic: usize§k: usize

Implementations§

impl Conv2d

pub fn apply(&self, x: &[f32], h: usize, w: usize) -> Vec<f32>

x: [ic, h, w] → [oc, h, w]. Banded im2col + GEMM: bands of output rows are lowered to a [rows·w, ic·k²] patch matrix and hit fcd_ops::gemm_nt (Accelerate/AMX on macOS, the portable blocked kernel elsewhere) — the band cap keeps the patch matrix ≤ ~128 MB at any image size.

Auto Trait Implementations§

impl Freeze for Conv2d

impl RefUnwindSafe for Conv2d

impl Send for Conv2d

impl Sync for Conv2d

impl Unpin for Conv2d

impl UnsafeUnpin for Conv2d

impl UnwindSafe for Conv2d

Blanket Implementations§

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

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more

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

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more

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

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

Mutably borrows from an owned value. Read more

impl<T> From<T> for T

fn from(t: T) -> T

Returns the argument unchanged.

impl<T> Instrument for T

fn instrument(self, span: Span) -> Instrumented<Self>

Instruments this type with the provided Span, returning an Instrumented wrapper. Read more

fn in_current_span(self) -> Instrumented<Self>

Instruments this type with the current Span, returning an Instrumented wrapper. Read more

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

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

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

type Error = Infallible

The type returned in the event of a conversion error.

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

Performs the conversion.

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

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

The type returned in the event of a conversion error.

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

Performs the conversion.

impl<T> WithSubscriber for T

fn with_subscriber<S>(self, subscriber: S) -> WithDispatch<Self>
where S: Into<Dispatch>,

Attaches the provided Subscriber to this type, returning a WithDispatch wrapper. Read more

fn with_current_subscriber(self) -> WithDispatch<Self>

Attaches the current default Subscriber to this type, returning a WithDispatch wrapper. Read more