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

Add flags to __all__ in tensorflow v1 API.

Browse Source 
We eagerly initialize __all__ variable in the API template and then we make more imports. This fix just adds required names to __all__.

PiperOrigin-RevId: 252469926
This commit is contained in:
Mihai Maruseac 2019-06-10 13:17:05 -07:00 committed by TensorFlower Gardener
parent 4b10957c79
commit 12f66fcf0a
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/api_template_v1.__init__.py
View File

@ -78,6 +78,8 @@ if '__all__' in vars():
from tensorflow.python.platform import flags # pylint: disable=g-import-not-at-top
# The 'app' module will be imported as part of the placeholder section above.
app.flags = flags # pylint: disable=undefined-variable
if '__all__' in vars():
vars()['__all__'].append('flags')
# Load all plugin libraries from site-packages/tensorflow-plugins if we are
# running under pip.