Allow Evolved Transformer number of decoder attention heads to exceed 16.
Allow Evolved Transformer number of decoder attention heads to exceed 16.

PiperOrigin-RevId: 239634336

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