Add optional variables argument to optimize(). The default behavior remains unchanged
Add optional variables argument to optimize(). The default behavior remains unchanged

PiperOrigin-RevId: 245977065

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