Add flag to debug nans.
Add flag to debug nans.

PiperOrigin-RevId: 244439471

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