Correct modalities for TPU eval after recent changes.
Correct modalities for TPU eval after recent changes.

PiperOrigin-RevId: 236380935

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