Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tf.config.experimental.enable_mlir_graph_optimization

View source on GitHub

Enables experimental MLIR-Based TensorFlow Compiler Optimizations.

tf.config.experimental.enable_mlir_graph_optimization()

DO NOT USE, DEV AND TESTING ONLY AT THE MOMENT.

NOTE: MLIR-Based TensorFlow Compiler is under active development and has missing features, please refrain from using. This API exists for development and testing only.

TensorFlow Compiler Optimizations are responsible general graph level optimizations that in the current stack mostly done by Grappler graph optimizers.