pub struct KdaState {
pub conv: Vec<f32>,
pub recurrent: Vec<f32>,
}Expand description
Per-sequence KDA state. Both buffers are FP32, read-modify-write.
Fields§
§conv: Vec<f32>[conv_dim, conv_kernel] FP32.
recurrent: Vec<f32>[heads, head_dim, head_dim] FP32, K-major.
Implementations§
Trait Implementations§
Auto Trait Implementations§
impl Freeze for KdaState
impl RefUnwindSafe for KdaState
impl Send for KdaState
impl Sync for KdaState
impl Unpin for KdaState
impl UnwindSafe for KdaState
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