I think that the developmemt of these optimizations is going to migrate over MLiR

  
      

      TensorFlow
  

  
    

tf.config.experimental.enable_mlir_graph_optimization 8

  Enables experimental MLIR-Based TensorFlow Compiler Optimizations.