tf.config.optimizer.set_jit

TensorFlow 1 version

View source on GitHub

Set if JIT compilation is enabled.

View aliases

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.config.optimizer.set_jit

tf.config.optimizer.set_jit(
    enabled
)

Note that optimizations are only applied to code that is compiled into a graph. In eager mode, which is the TF2 API default, that means only code that is defined under a tf.function decorator.

Args
`enabled`	Whether to enable JIT compilation.

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 2021-02-18 UTC.

tf.config.optimizer.set_jit

View aliases

Args