tf.config.optimizer.set_jit

Set if JIT compilation is enabled.

Compat aliases for migration

tf.compat.v1.config.optimizer.set_jit, tf.compat.v2.config.optimizer.set_jit

tf.config.optimizer.set_jit(
    enabled
)