megatron.model.positional_embeddings# Description Functions apply_rotary_emb(xq, xk, freqs_cis[, ...]) precompute_freqs_cis(dim, end[, theta, ...]) reshape_for_broadcast(freqs_cis, x)