pub struct Pools {
pub keys: Vec<f32>,
pub indices: Vec<i32>,
pub valid: Vec<u8>,
pub n_pools: usize,
}Expand description
The compressed k-pool candidates.
Fields§
§keys: Vec<f32>[n_pools, index_head_dim] — softmax-weighted average of the pool’s keys.
indices: Vec<i32>[n_pools, index_kpool] — raw token index per slot, INVALID where the slot is not real.
valid: Vec<u8>[n_pools] — a pool counts only when EVERY slot is valid.
n_pools: usizeAuto Trait Implementations§
impl Freeze for Pools
impl RefUnwindSafe for Pools
impl Send for Pools
impl Sync for Pools
impl Unpin for Pools
impl UnwindSafe for Pools
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