Struct GraphEdge 

pub struct GraphEdge {
    pub edge_type: GraphEdgeType,
    pub source: usize,
    pub target: usize,
    pub self_attention: Option<SelfAttentionEdge>,
    pub data_flow: Option<DataFlowEdge>,
    pub residual: Option<ResidualEdge>,
}

Graph edge wrapper

Fields

edge_type: GraphEdgeType

Edge type

source: usize

Source node ID

target: usize

Target node ID

self_attention: Option<SelfAttentionEdge>

Optional self-attention data

data_flow: Option<DataFlowEdge>

Optional data flow data

residual: Option<ResidualEdge>

Optional residual data

Implementations

impl GraphEdge

pub fn self_attention( source: usize, target: usize, weight: f64, head: usize, layer: usize, ) -> Self

Create a self-attention edge

pub fn data_flow( source: usize, target: usize, operation: DataFlowOp, layer: usize, ) -> Self

Create a data flow edge

pub fn residual( source: usize, target: usize, layer: usize, skip_type: SkipType, ) -> Self

Create a residual edge

pub fn get_self_attention(&self) -> Option<&SelfAttentionEdge>

Get the self-attention data if applicable

pub fn get_data_flow(&self) -> Option<&DataFlowEdge>

Get the data flow info if applicable

pub fn get_residual(&self) -> Option<&ResidualEdge>

Get the residual info if applicable

pub fn layer(&self) -> usize

Get the layer number

pub fn self_attention_with_message( source: usize, target: usize, weight: f64, head: usize, layer: usize, message: DenseTensor, ) -> Self

Create a self-attention edge with message tensor

pub fn data_flow_with_message( source: usize, target: usize, operation: DataFlowOp, layer: usize, message: DenseTensor, ) -> Self

Create a data flow edge with message tensor

pub fn residual_with_tensor( source: usize, target: usize, layer: usize, skip_type: SkipType, residual: DenseTensor, ) -> Self

Create a residual edge with tensor

pub fn message(&self) -> Option<&DenseTensor>

Get the message tensor from this edge (if any)

pub fn set_message(&mut self, message: DenseTensor) -> bool

Set the message tensor on this edge

pub fn self_attention_with_qkv( source: usize, target: usize, weight: f64, head: usize, layer: usize, q_proj: DenseTensor, k_proj: DenseTensor, v_proj: DenseTensor, ) -> Self

Create a self-attention edge with separate Q, K, V projections

pub fn get_qkv( &self, ) -> (Option<&DenseTensor>, Option<&DenseTensor>, Option<&DenseTensor>)

Get Q/K/V projections if available (SelfAttention edges only)

pub fn has_qkv(&self) -> bool

Check if this edge has complete QKV projections

pub fn key_proj(&self) -> Option<&DenseTensor>

Get the key projection (SelfAttention edges only)

pub fn value_proj(&self) -> Option<&DenseTensor>

Get the value projection (SelfAttention edges only)

pub fn compute_attention_score(&self, d_k: f64) -> Option<f64>

Compute attention score using Q and K projections (SelfAttention edges only)

Trait Implementations

impl Clone for GraphEdge

fn clone(&self) -> GraphEdge

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for GraphEdge

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.