Remove unintended log message and a few other cleanups.
Remove unintended log message and a few other cleanups.

PiperOrigin-RevId: 230637260

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