pub struct LmHeadOverlay {
pub rows: DevicePtr,
pub ids_dev: DevicePtr,
pub n_override: u32,
}Expand description
Distinct output-projection overlay for an untied lm_head. Rows are recomputed
into logits by a dot(hidden, row) per overridden id (no slot_map).
Fields§
§rows: DevicePtr§ids_dev: DevicePtr§n_override: u32Trait Implementations§
Auto Trait Implementations§
impl Freeze for LmHeadOverlay
impl RefUnwindSafe for LmHeadOverlay
impl Send for LmHeadOverlay
impl Sync for LmHeadOverlay
impl Unpin for LmHeadOverlay
impl UnwindSafe for LmHeadOverlay
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