Add source file links to objects imported from keras-preprocessing.

PiperOrigin-RevId: 340732811
Change-Id: If0f9eb3de85637e2cdffc7552de7cc1dadd067a8

tensorflow/tools/docs/base_dir.py

@@ -21,6 +21,7 @@ from __future__ import print_function
 import distutils
 from os import path
 
+import keras_preprocessing
 import tensorboard
 import tensorflow as tf
 import tensorflow_estimator
@@ -33,18 +34,21 @@ def get_base_dirs_and_prefixes(code_url_prefix):
   if distutils.version.LooseVersion(tf.__version__) >= "2.2":
     base_dirs = [
         base_dir,
+        path.dirname(keras_preprocessing.__file__),
         path.dirname(tensorboard.__file__),
         path.dirname(tensorflow_estimator.__file__),
     ]
   else:
     base_dirs = [
         path.normpath(path.join(base_dir, "../tensorflow_core")),
+        path.dirname(keras_preprocessing.__file__),
         path.dirname(tensorboard.__file__),
         path.dirname(tensorflow_estimator.__file__),
     ]
 
   code_url_prefixes = (
       code_url_prefix,
+      "https://github.com/keras-team/keras-preprocessing/tree/master/keras_preprocessing",
       "https://github.com/tensorflow/tensorboard/tree/master/tensorboard",
       "https://github.com/tensorflow/estimator/tree/master/tensorflow_estimator",
   )

tensorflow/tools/docs/generate2.py

@@ -239,6 +239,8 @@ def build_docs(output_dir, code_url_prefix, search_hints, gen_report):
           "python/ops/nn_impl.py",
       "tf/keras/Model.md":
           "tensorflow/python/keras/engine/training.py",
+      "tf/keras/preprocessing/image/random_brightness.md":
+          "keras_preprocessing/image/affine_transformations.py"
   }
 
   all_passed = True