Fix PY3 compatibility bug with map() input.
Fix PY3 compatibility bug with map() input.

PiperOrigin-RevId: 276357497

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