Add support for quantizing `batch_major_attention.MultiHeadedAttention`.
Add support for quantizing `batch_major_attention.MultiHeadedAttention`.

PiperOrigin-RevId: 374260066

tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo