tf.executing_eagerly | TensorFlow v2.16.1 (original) (raw)
tf.executing_eagerly
Stay organized with collections Save and categorize content based on your preferences.
Checks whether the current thread has eager execution enabled.
tf.executing_eagerly()
Used in the notebooks
Eager execution is enabled by default and this API returns True
in most of cases. However, this API might return False
in the following use cases.
- Executing inside tf.function, unless under tf.init_scope ortf.config.run_functions_eagerly(True) is previously called.
- Executing inside a transformation function for
tf.dataset
. - tf.compat.v1.disable_eager_execution() is called.
General case:
print(tf.executing_eagerly())
True
Inside tf.function:
@tf.function
def fn():
with tf.init_scope():
print(tf.executing_eagerly())
print(tf.executing_eagerly())
fn()
True
False
Inside tf.function after tf.config.run_functions_eagerly(True) is called:
tf.config.run_functions_eagerly(True)
@tf.function
def fn():
with tf.init_scope():
print(tf.executing_eagerly())
print(tf.executing_eagerly())
fn()
True
True
tf.config.run_functions_eagerly(False)
Inside a transformation function for tf.dataset
:
def data_fn(x):
print(tf.executing_eagerly())
return x
dataset = tf.data.Dataset.range(100)
dataset = dataset.map(data_fn)
False
Returns |
---|
True if the current thread has eager execution enabled. |
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.
Last updated 2024-04-26 UTC.