pub enum TensorRole {
Show 22 variants
TokenEmbd,
Output,
OutputNorm,
AttnNorm,
AttnQ,
AttnK,
AttnV,
AttnQkv,
AttnOut,
AttnQNorm,
AttnKNorm,
AttnPostNorm,
FfnNorm,
FfnGate,
FfnUp,
FfnDown,
FfnPostNorm,
FfnGateInp,
FfnGateExps,
FfnUpExps,
FfnDownExps,
RopeFreqs,
}Expand description
Logical weight / activation tensor identity inside a decoder block.
Variants§
TokenEmbd
Output
OutputNorm
AttnNorm
AttnQ
AttnK
AttnV
AttnQkv
AttnOut
AttnQNorm
AttnKNorm
AttnPostNorm
FfnNorm
FfnGate
FfnUp
FfnDown
FfnPostNorm
FfnGateInp
FfnGateExps
FfnUpExps
FfnDownExps
RopeFreqs
Implementations§
Trait Implementations§
Source§impl Clone for TensorRole
impl Clone for TensorRole
Source§fn clone(&self) -> TensorRole
fn clone(&self) -> TensorRole
Returns a duplicate of the value. Read more
1.0.0 (const: unstable) · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source. Read moreimpl Copy for TensorRole
Source§impl Debug for TensorRole
impl Debug for TensorRole
impl Eq for TensorRole
Source§impl Hash for TensorRole
impl Hash for TensorRole
Source§impl PartialEq for TensorRole
impl PartialEq for TensorRole
impl StructuralPartialEq for TensorRole
Auto Trait Implementations§
impl Freeze for TensorRole
impl RefUnwindSafe for TensorRole
impl Send for TensorRole
impl Sync for TensorRole
impl Unpin for TensorRole
impl UnsafeUnpin for TensorRole
impl UnwindSafe for TensorRole
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