Update Migrate single-worker distributed training (migrating to TF2)
Update Migrate single-worker distributed training (migrating to TF2)

PiperOrigin-RevId: 391811492

tensorflow/docs,tensorflow/docs,tensorflow/docs