Function activation_wire_bytes_with_state_flags 

pub fn activation_wire_bytes_with_state_flags(
    dtype: WireActivationDType,
    token_count: i32,
    n_embd: i32,
    state_flag_bits: i32,
) -> Result<usize>