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

[tf.data] Warn users that the `tf.config.experimental_run_functions_eagerly` option does not apply to tf.data functions. 

PiperOrigin-RevId: 308467977
Change-Id: I4f31f088191317cfb7cc8051291882f17e7fef79
This commit is contained in:
Jiri Simsa 2020-04-25 22:36:11 -07:00 committed by TensorFlower Gardener
parent e56abe5a08
commit 64c32cf510
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/python/data/ops/dataset_ops.py
View File

@ -42,6 +42,7 @@ from tensorflow.python.data.util import random_seed
from tensorflow.python.data.util import structure
from tensorflow.python.data.util import traverse
from tensorflow.python.eager import context
from tensorflow.python.eager import def_function
from tensorflow.python.eager import function as eager_function
from tensorflow.python.framework import auto_control_deps
from tensorflow.python.framework import auto_control_deps_utils as acd_utils
@ -3267,6 +3268,12 @@ class StructuredFunctionWrapper(object):
_warn_if_collections(transformation_name)
else:
if def_function.functions_run_eagerly():
warnings.warn(
"Even though the tf.config.experimental_run_functions_eagerly "
"option is set, this option does not apply to tf.data functions. "
"tf.data functions are still traced and executed as graphs.")
defun_kwargs.update({"func_name": func_name})
defun_kwargs.update({"_tf_data_function": True})