tf.config.optimizer.set_jit

TensorFlow 1 version View source on GitHub

Set if JIT compilation is enabled.

Used in the notebooks

Used in the tutorials

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.

enabled Whether to enable JIT compilation.