From 3006330ea0c3e88651195ac7c7de654291377ebb Mon Sep 17 00:00:00 2001 From: Rohan Jain Date: Wed, 4 Mar 2020 21:14:55 -0800 Subject: [PATCH] Disabling v2 in the disable_v2_behavior() method PiperOrigin-RevId: 299012811 Change-Id: I311f4b8a6ecbabb658717fff248bea072ff7a366 --- tensorflow/python/compat/v2_compat.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tensorflow/python/compat/v2_compat.py b/tensorflow/python/compat/v2_compat.py index c563a215c10..eac841fb2fe 100644 --- a/tensorflow/python/compat/v2_compat.py +++ b/tensorflow/python/compat/v2_compat.py @@ -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()