Adjust books to account for changed to the tensorflow_docs api_generator.
Adjust books to account for changed to the tensorflow_docs api_generator.

PiperOrigin-RevId: 331859959

tensorflow/tfx,tensorflow/tfx