Function reference_value_aggregation 

pub fn reference_value_aggregation(
    probabilities: &[f32],
    values: &[f32],
    head_dim: usize,
) -> Result<Vec<f32>>

Compute reference value aggregation from attention probabilities.