Update Adafactor optimizer to new signature.
Update Adafactor optimizer to new signature.

PiperOrigin-RevId: 260010526

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