megatron.model.positional_embeddings.reshape_for_broadcast

megatron.model.positional_embeddings.reshape_for_broadcast(freqs_cis: Tensor, x: Tensor) → Tensor