megatron.model.positional_embeddings.apply_rotary_emb

megatron.model.positional_embeddings.apply_rotary_emb(xq: Tensor, xk: Tensor, freqs_cis: Tensor, position_ids: Tensor | None = None) → tuple[torch.Tensor, torch.Tensor]