pub struct DraftGeom {
pub d_inner: usize,
pub n_head: usize,
pub n_head_kv: usize,
pub out_up: GpuTensor,
}Expand description
Draft-head geometry override for a distilled (narrower) student block.
Fields§
§d_inner: usize§n_head: usize§n_head_kv: usize§out_up: GpuTensorAuto Trait Implementations§
impl Freeze for DraftGeom
impl RefUnwindSafe for DraftGeom
impl Send for DraftGeom
impl Sync for DraftGeom
impl Unpin for DraftGeom
impl UnsafeUnpin for DraftGeom
impl UnwindSafe for DraftGeom
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