tf.config.optimizer.set_jit

View source on GitHub

Set if JIT compilation is enabled.

Aliases:

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

tf.config.optimizer.set_jit(enabled)

Args:

• enabled: Whether to enable JIT compilation.