pub struct SamNeckWeights {
pub conv1_w: Vec<f32>,
pub ln1_g: Vec<f32>,
pub ln1_b: Vec<f32>,
pub conv2_w: Vec<f32>,
pub ln2_g: Vec<f32>,
pub ln2_b: Vec<f32>,
pub embed_dim: usize,
pub out_chans: usize,
pub eps: f32,
}Expand description
Weights for the four neck layers, kept on the host because rlx-ir doesn’t have f32 forward Conv2d (and 3×3 padding=1 doesn’t reduce to matmul).
Fields§
§conv1_w: Vec<f32>§ln1_g: Vec<f32>§ln1_b: Vec<f32>§conv2_w: Vec<f32>§ln2_g: Vec<f32>§ln2_b: Vec<f32>§embed_dim: usize§out_chans: usize§eps: f32Auto Trait Implementations§
impl Freeze for NeckWeights
impl RefUnwindSafe for NeckWeights
impl Send for NeckWeights
impl Sync for NeckWeights
impl Unpin for NeckWeights
impl UnsafeUnpin for NeckWeights
impl UnwindSafe for NeckWeights
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
impl<ST, DT> CastableFrom<ST, Initialized, Initialized> for DT
impl<ST, DT> CastableFrom<ST, Uninit, Uninit> for DT
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more