Add unit test for ASR input, and Area Attention.
Add unit test for ASR input, and Area Attention.

PiperOrigin-RevId: 259582396

tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor