experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Set the library name based on the version of TensorFlow

Browse Source 
PiperOrigin-RevId: 263035092
This commit is contained in:
Yash Katariya 2019-08-12 16:59:21 -07:00 committed by TensorFlower Gardener
parent 41e3677707
commit 8693a9eaa3
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/tools/docs/generate2.py
View File

@ -250,8 +250,13 @@ def build_docs(output_dir, code_url_prefix, search_hints=True):
"https://github.com/tensorflow/estimator/tree/master/tensorflow_estimator", "https://github.com/tensorflow/estimator/tree/master/tensorflow_estimator",
) )
if LooseVersion(tf.__version__) < LooseVersion('2'):
root_title = 'TensorFlow'
elif LooseVersion(tf.__version__) >= LooseVersion('2'):
root_title = 'TensorFlow 2.0'
doc_generator = generate_lib.DocGenerator( doc_generator = generate_lib.DocGenerator(
root_title="TensorFlow 2.0 Preview", root_title=root_title,
py_modules=[("tf", tf)], py_modules=[("tf", tf)],
base_dir=base_dirs, base_dir=base_dirs,
search_hints=search_hints, search_hints=search_hints,