pub enum GateWidth {
PerHead,
PerElement,
}Expand description
How many gate values one token gets, and therefore how they broadcast over the attention output.
Variants§
PerHead
n_head values; each scales its head’s head_dim channels.
PerElement
n_head * head_dim values, one per channel.
Trait Implementations§
impl Copy for GateWidth
impl Eq for GateWidth
impl StructuralPartialEq for GateWidth
Auto Trait Implementations§
impl Freeze for GateWidth
impl RefUnwindSafe for GateWidth
impl Send for GateWidth
impl Sync for GateWidth
impl Unpin for GateWidth
impl UnsafeUnpin for GateWidth
impl UnwindSafe for GateWidth
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
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
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