"compiled inside the tpu cluster will also be replicated across "
            "multiple cpu devices."),
       Flag("tf_dump_graphs_in_tfg", &use_tfg_graph_dumper,
            "When tf_dump_graphs_in_tfg is true, graphs after transformations "
            "are dumped in MLIR TFG dialect and not in GraphDef"),
       Flag("tf_mlir_enable_tpu_variable_runtime_reformatting_pass",