pub struct FlowState {Show 13 fields
pub rope_cos: Option<HirNodeId>,
pub rope_sin: Option<HirNodeId>,
pub zero_beta: Option<HirNodeId>,
pub embed_weight: Option<HirNodeId>,
pub hidden_shape: Option<Shape>,
pub decode: Option<DecodeBindings>,
pub residual_skip: Option<HirNodeId>,
pub residual_shape: Option<Shape>,
pub streams: HashMap<String, FlowValue>,
pub inputs: HashMap<String, (HirNodeId, Shape)>,
pub named: HashMap<String, HirNodeId>,
pub gdn: Option<GdnInputSlots>,
pub loaded_params: HashMap<String, HirNodeId>,
}Expand description
Cross-stage shared handles (RoPE tables, zero-beta, tied embed, …).
Fields§
§rope_cos: Option<HirNodeId>§rope_sin: Option<HirNodeId>§zero_beta: Option<HirNodeId>§embed_weight: Option<HirNodeId>§decode: Option<DecodeBindings>§residual_skip: Option<HirNodeId>§residual_shape: Option<Shape>§streams: HashMap<String, FlowValue>Named tensor streams (img, txt, …) for multi-stream models.
inputs: HashMap<String, (HirNodeId, Shape)>Graph inputs beyond the primary tensor flow (encoder, temb, …).
named: HashMap<String, HirNodeId>Named scalar/tensor node refs (RoPE tables, mod params, carry state, …).
gdn: Option<GdnInputSlots>Last-published GDN q/k/v/g/beta handles for crate::blocks::GdnScanStage.
loaded_params: HashMap<String, HirNodeId>Reuse param nodes when multiple stages in one layer load the same key
(e.g. crate::blocks::LlamaKvTapStage + fused decoder).
Trait Implementations§
Auto Trait Implementations§
impl Freeze for FlowState
impl RefUnwindSafe for FlowState
impl Send for FlowState
impl Sync for FlowState
impl Unpin for FlowState
impl UnsafeUnpin for FlowState
impl UnwindSafe for FlowState
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