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

4
tensorflow/api_template.__init__.py
View File

@ -41,6 +41,10 @@ import sys as _sys
from tensorflow.python.tools import module_util as _module_util
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
# WRAPPER_PLACEHOLDER