pub struct LinearAttentionState {
pub conv1d_state: DevicePtr,
pub gdn_state: DevicePtr,
}Expand description
2026-09-25: Conv and GDN state of one linear-attention layer, carried from token to token. The caller allocates and zeroes it.
Fields§
§conv1d_state: DevicePtr2026-09-25: FP32 [qkv_total_lin, conv_kernel_size].
gdn_state: DevicePtr2026-09-25: FP32 [num_v_heads_lin, k_head_dim_lin, v_head_dim_lin].
Auto Trait Implementations§
impl Freeze for LinearAttentionState
impl RefUnwindSafe for LinearAttentionState
impl Send for LinearAttentionState
impl Sync for LinearAttentionState
impl Unpin for LinearAttentionState
impl UnwindSafe for LinearAttentionState
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