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

[Tensorflow Metrics] Add tracker for enable_v2_behavior() and disable_v2_behavior.

Browse Source 
DOC=https://docs.google.com/document/d/1Xk21sJBxtJAUvvLtQSYopiBdHe_8wUzv4kiICVYz8Jg/edit#heading=h.ng49b3y8n8x4

PiperOrigin-RevId: 288577048
Change-Id: Icdeb43ed4ec4606fdcd2da2930d5018042e72a4c
This commit is contained in:
A. Unique TensorFlower 2020-01-07 14:48:25 -08:00 committed by TensorFlower Gardener
parent 9cf866b53f
commit a146fec654
2 changed files with 9 additions and 1 deletions
tensorflow/python/compat
BUILDv2_compat.py

1
tensorflow/python/compat/BUILD
View File

@ -15,6 +15,7 @@ py_library(
"//tensorflow/python:control_flow_v2_toggles",
"//tensorflow/python:tf2",
"//tensorflow/python:util",
"//tensorflow/python/eager:monitoring",
],
)

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

@ -25,6 +25,7 @@ from tensorflow.python.data.experimental.ops import random_ops
from tensorflow.python.data.experimental.ops import readers as exp_readers
from tensorflow.python.data.ops import dataset_ops
from tensorflow.python.data.ops import readers
from tensorflow.python.eager import monitoring
from tensorflow.python.framework import ops
from tensorflow.python.framework import tensor_shape
from tensorflow.python.ops import control_flow_v2_toggles
@ -32,6 +33,11 @@ from tensorflow.python.ops import variable_scope
from tensorflow.python.util.tf_export import tf_export
# Metrics to track the status of v2_behavior
_v2_behavior_usage_gauge = monitoring.BoolGauge(
"/tensorflow/version/v2_behavior",
"whether v2_behavior is enabled or disabled", "status")
@tf_export(v1=["enable_v2_behavior"])
def enable_v2_behavior():
@ -45,6 +51,7 @@ def enable_v2_behavior():
This function is called in the main TensorFlow `__init__.py` file, user should
not need to call it, except during complex migrations.
"""
_v2_behavior_usage_gauge.get_cell("enable").set(True)
# TF2 behavior is enabled if either 1) enable_v2_behavior() is called or
# 2) the TF2_BEHAVIOR=1 environment variable is set. In the latter case,
# the modules below independently check if tf2.enabled().
@ -82,7 +89,7 @@ def disable_v2_behavior():
User can call this function to disable 2.x behavior during complex migrations.
"""
tf2.disable()
_v2_behavior_usage_gauge.get_cell("disable").set(True)
ops.disable_eager_execution()
tensor_shape.disable_v2_tensorshape() # Also switched by tf2
variable_scope.disable_resource_variables()