Searched defs:freqs_sin (Results 1 – 3 of 3) sorted by relevance
27 const at::Tensor& freqs_sin) { in rotary_embedding_impl()92 at::Tensor freqs_sin = in test_reference() local
33 const ValueRef freqs_sin, in add_rotary_embedding_node()
437 def forward(self, x, freqs_cos, freqs_sin, input_pos=None): # x: 1xN argument