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

Make sure that TensorFlow code can use tf2.enabled() at import time.

Browse Source 
Fixes #36700.

PiperOrigin-RevId: 302924977
Change-Id: I22991018941dcb1ad9a862320f1effaa5238218a
This commit is contained in:
Martin Wicke 2020-03-25 10:55:17 -07:00 committed by TensorFlower Gardener
parent ac191b5591
commit 35a382295a
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
tensorflow/api_template.__init__.py
View File

@ -13,16 +13,16 @@
# limitations under the License. # limitations under the License.
# ============================================================================== # ==============================================================================
""" """
Top-level module of TensorFlow. By convention, we refer to this module as Top-level module of TensorFlow. By convention, we refer to this module as
`tf` instead of `tensorflow`, following the common practice of importing `tf` instead of `tensorflow`, following the common practice of importing
TensorFlow via the command `import tensorflow as tf`. TensorFlow via the command `import tensorflow as tf`.
The primary function of this module is to import all of the public TensorFlow The primary function of this module is to import all of the public TensorFlow
interfaces into a single place. The interfaces themselves are located in interfaces into a single place. The interfaces themselves are located in
sub-modules, as described below. sub-modules, as described below.
Note that the file `__init__.py` in the TensorFlow source code tree is actually Note that the file `__init__.py` in the TensorFlow source code tree is actually
only a placeholder to enable test cases to run. The TensorFlow build replaces only a placeholder to enable test cases to run. The TensorFlow build replaces
this file with a file generated from [`api_template.__init__.py`](https://www.github.com/tensorflow/tensorflow/blob/master/tensorflow/api_template.__init__.py) this file with a file generated from [`api_template.__init__.py`](https://www.github.com/tensorflow/tensorflow/blob/master/tensorflow/api_template.__init__.py)
""" """
@ -41,6 +41,10 @@ import sys as _sys
from tensorflow.python.tools import module_util as _module_util from tensorflow.python.tools import module_util as _module_util
from tensorflow.python.util.lazy_loader import LazyLoader as _LazyLoader from tensorflow.python.util.lazy_loader import LazyLoader as _LazyLoader
# Make sure code inside the TensorFlow codebase can use tf2.enabled() at import.
from tensorflow.python import tf2 as _tf2
_tf2.enable()
# API IMPORTS PLACEHOLDER # API IMPORTS PLACEHOLDER
# WRAPPER_PLACEHOLDER # WRAPPER_PLACEHOLDER