Fix mlperf numpy array logging issue.
Fix mlperf numpy array logging issue.

PiperOrigin-RevId: 218706375

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