Struct GraphRoPEAttention 

pub struct GraphRoPEAttention { /* private fields */ }

Graph RoPE attention (Rotary Position Embeddings for graphs)

Implementations

impl GraphRoPEAttention

pub fn instance_of<V: NapiRaw>(env: Env, value: V) -> Result<bool>

impl GraphRoPEAttention

pub fn into_reference( val: GraphRoPEAttention, env: Env, ) -> Result<Reference<GraphRoPEAttention>>

pub fn into_instance( self, env: Env, ) -> Result<ClassInstance<GraphRoPEAttention>>

impl GraphRoPEAttention

pub fn new(config: RoPEConfig) -> Self

Create a new Graph RoPE attention instance

Arguments

• config - RoPE configuration

pub fn simple(dim: u32, max_position: u32) -> Self

Create with simple parameters

pub fn compute( &self, query: Float32Array, keys: Vec<Float32Array>, values: Vec<Float32Array>, ) -> Result<Float32Array>

Compute attention without positional encoding

pub fn compute_with_positions( &self, query: Float32Array, keys: Vec<Float32Array>, values: Vec<Float32Array>, query_position: u32, key_positions: Vec<u32>, ) -> Result<Float32Array>

Compute attention with graph positions

Arguments

• query - Query vector
• keys - Array of key vectors
• values - Array of value vectors
• query_position - Position of query node
• key_positions - Positions of key nodes (e.g., hop distances)

pub fn apply_rotary(&self, vector: Float32Array, position: u32) -> Float32Array

Apply rotary embedding to a vector

pub fn distance_to_position(distance: u32, max_distance: u32) -> u32

Convert graph distance to position bucket

pub fn dim(&self) -> u32

Get the dimension

pub fn max_position(&self) -> u32

Get the max position

Trait Implementations

impl FromNapiMutRef for GraphRoPEAttention

unsafe fn from_napi_mut_ref( env: napi_env, napi_val: napi_value, ) -> Result<&'static mut Self>

Safety

impl FromNapiRef for GraphRoPEAttention

unsafe fn from_napi_ref( env: napi_env, napi_val: napi_value, ) -> Result<&'static Self>

Safety

impl FromNapiValue for &GraphRoPEAttention

unsafe fn from_napi_value(env: napi_env, napi_val: napi_value) -> Result<Self>

Safety

fn from_unknown(value: JsUnknown) -> Result<Self, Error>

impl FromNapiValue for &mut GraphRoPEAttention

unsafe fn from_napi_value(env: napi_env, napi_val: napi_value) -> Result<Self>

Safety

fn from_unknown(value: JsUnknown) -> Result<Self, Error>

impl ObjectFinalize for GraphRoPEAttention

fn finalize(self, env: Env) -> Result<(), Error>

impl ToNapiValue for GraphRoPEAttention

unsafe fn to_napi_value( env: napi_env, val: GraphRoPEAttention, ) -> Result<napi_value>

Safety

impl TypeName for &GraphRoPEAttention

fn type_name() -> &'static str

fn value_type() -> ValueType

impl TypeName for &mut GraphRoPEAttention

fn type_name() -> &'static str

fn value_type() -> ValueType

impl TypeName for GraphRoPEAttention

fn type_name() -> &'static str

fn value_type() -> ValueType

impl ValidateNapiValue for &GraphRoPEAttention

unsafe fn validate(env: napi_env, napi_val: napi_value) -> Result<napi_value>

Safety

impl ValidateNapiValue for &mut GraphRoPEAttention

unsafe fn validate(env: napi_env, napi_val: napi_value) -> Result<napi_value>

Safety