Remove deprecated ps_devices argument.
Remove deprecated ps_devices argument.

PiperOrigin-RevId: 219215973

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