pub struct SamLayer<'a> {Show 14 fields
pub ln1_w: &'a [f32],
pub ln1_b: &'a [f32],
pub qkv: &'a [f32],
pub qkv_b: &'a [f32],
pub proj: &'a [f32],
pub proj_b: &'a [f32],
pub rel: &'a [f32],
pub ln2_w: &'a [f32],
pub ln2_b: &'a [f32],
pub fc1: &'a [f32],
pub fc1_b: &'a [f32],
pub fc2: &'a [f32],
pub fc2_b: &'a [f32],
pub window: usize,
}Expand description
One SAM block for sam_encode. rel is [2w−1, hd] relative positions,
h then w (window blocks; the global ones take SamArgs::rel_global).
Fields§
§ln1_w: &'a [f32]§ln1_b: &'a [f32]§qkv: &'a [f32]§qkv_b: &'a [f32]§proj: &'a [f32]§proj_b: &'a [f32]§rel: &'a [f32]§ln2_w: &'a [f32]§ln2_b: &'a [f32]§fc1: &'a [f32]§fc1_b: &'a [f32]§fc2: &'a [f32]§fc2_b: &'a [f32]§window: usizeAuto Trait Implementations§
impl<'a> Forget for SamLayer<'a>
impl<'a> Freeze for SamLayer<'a>
impl<'a> Move for SamLayer<'a>
impl<'a> RefUnwindSafe for SamLayer<'a>
impl<'a> Send for SamLayer<'a>
impl<'a> Sync for SamLayer<'a>
impl<'a> Unpin for SamLayer<'a>
impl<'a> UnsafeUnpin for SamLayer<'a>
impl<'a> UnwindSafe for SamLayer<'a>
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