Fix minor typo in TensorFlow "basic training loops" documentation.
Fix minor typo in TensorFlow "basic training loops" documentation.

PiperOrigin-RevId: 364618382

tensorflow/docs,tensorflow/docs,tensorflow/docs