tf.config.optimizer.set_jit

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]
)

Used in the notebooks

Used in the tutorials
Classifying CIFAR-10 with XLA

Deprecated: SOME ARGUMENT VALUES ARE DEPRECATED: (jit_config=True). They will be removed in a future version. Instructions for updating: True setting is deprecated, use autoclustering instead.

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.