pass tpu_job_name to TPUConfig.
pass tpu_job_name to TPUConfig.

PiperOrigin-RevId: 240367576

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