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

Disabling v2 in the disable_v2_behavior() method

Browse Source 
PiperOrigin-RevId: 299012811
Change-Id: I311f4b8a6ecbabb658717fff248bea072ff7a366
This commit is contained in:
Rohan Jain 2020-03-04 21:14:55 -08:00 committed by TensorFlower Gardener
parent 3b6251b6cc
commit 3006330ea0
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/python/compat/v2_compat.py
View File

@ -90,6 +90,7 @@ def disable_v2_behavior():
User can call this function to disable 2.x behavior during complex migrations.
"""
_v2_behavior_usage_gauge.get_cell("disable").set(True)
tf2.disable()
ops.disable_eager_execution()
tensor_shape.disable_v2_tensorshape() # Also switched by tf2
variable_scope.disable_resource_variables()