pub struct LutAttentionSpec {}Expand description
Let C be lut attention of Q over K and V with H heads KV groups [up to P] with table E and scale M shifted right by S — single-query attention over Int tensors whose softmax is an exp table: per head, scores q · k_j, their
maximum, weights E[clamp(((max − s_j) · M) >> S, 0, table_len − 1)] and the context ⌊Σ w_j · v_j / Σ w_j⌋.
The shapes and pos are AttentionSpec’s.
Fields§
§q: Symbol§kc: Symbol§vc: Symbol§table: Symbol§out: Symbol§h: usize§kv: usize§hd: usize§l: usize§table_len: usize§pos: Option<RowIndex>§m: i64§s: u32Trait Implementations§
Source§impl Clone for LutAttentionSpec
impl Clone for LutAttentionSpec
Source§fn clone(&self) -> LutAttentionSpec
fn clone(&self) -> LutAttentionSpec
Returns a duplicate of the value. Read more
1.0.0 · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source. Read moreSource§impl Debug for LutAttentionSpec
impl Debug for LutAttentionSpec
Source§impl PartialEq for LutAttentionSpec
impl PartialEq for LutAttentionSpec
impl Eq for LutAttentionSpec
impl StructuralPartialEq for LutAttentionSpec
Auto Trait Implementations§
impl Freeze for LutAttentionSpec
impl RefUnwindSafe for LutAttentionSpec
impl Send for LutAttentionSpec
impl Sync for LutAttentionSpec
impl Unpin for LutAttentionSpec
impl UnsafeUnpin for LutAttentionSpec
impl UnwindSafe for LutAttentionSpec
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,
§impl<T> Downcast for Twhere
T: Any,
impl<T> Downcast for Twhere
T: Any,
§fn into_any(self: Box<T>) -> Box<dyn Any>
fn into_any(self: Box<T>) -> Box<dyn Any>
Convert
Box<dyn Trait> (where Trait: Downcast) to Box<dyn Any>. Box<dyn Any> can
then be further downcast into Box<ConcreteType> where ConcreteType implements Trait.§fn into_any_rc(self: Rc<T>) -> Rc<dyn Any>
fn into_any_rc(self: Rc<T>) -> Rc<dyn Any>
Convert
Rc<Trait> (where Trait: Downcast) to Rc<Any>. Rc<Any> can then be
further downcast into Rc<ConcreteType> where ConcreteType implements Trait.§fn as_any(&self) -> &(dyn Any + 'static)
fn as_any(&self) -> &(dyn Any + 'static)
Convert
&Trait (where Trait: Downcast) to &Any. This is needed since Rust cannot
generate &Any’s vtable from &Trait’s.§fn as_any_mut(&mut self) -> &mut (dyn Any + 'static)
fn as_any_mut(&mut self) -> &mut (dyn Any + 'static)
Convert
&mut Trait (where Trait: Downcast) to &Any. This is needed since Rust cannot
generate &mut Any’s vtable from &mut Trait’s.§impl<T> DowncastSync for T
impl<T> DowncastSync for T
§impl<Q, K> Equivalent<K> for Q
impl<Q, K> Equivalent<K> for Q
§fn equivalent(&self, key: &K) -> bool
fn equivalent(&self, key: &K) -> bool
Checks if this value is equivalent to the given key. Read more
§impl<Q, K> Equivalent<K> for Q
impl<Q, K> Equivalent<K> for Q
§fn equivalent(&self, key: &K) -> bool
fn equivalent(&self, key: &K) -> bool
Compare self to
key and return true if they are equal.