tf.config.optimizer.set_jit

View source on GitHub

Configure JIT compilation. (deprecated argument values)

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: Union[bool, str]
)

Note: compilation is only applied to code that is compiled into a graph (in TF2 that's only a code inside tf.function).

Args
enabled	JIT compilation configuration. Possible values: "autoclustering" (True is a deprecated alias): perform autoclustering (automatically identify and compile clusters of nodes) on all graphs using XLA. False: do not automatically compile any graphs.