"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",
            &enable_tpu_variable_runtime_reformatting_pass,
            "Enables TPUVariableRuntimeReformatting pass for MLIR-Based "

      mlir::TFTPU::CreateExtractTPUCopyWithDynamicShapeOpPass());
  pm.addNestedPass<FuncOp>(
      mlir::TFTPU::CreateTPUColocateCompositeResourceOps());
  if (tensorflow::GetMlirCommonFlags()
          ->tf_mlir_enable_tpu_variable_runtime_reformatting_pass) {
    pm.addPass(mlir::TFTPU::CreateTPUVariableRuntimeReformattingPass());
  }
}

void AddNonTPULowerClusterToRuntimeOpsPassPipeline(
    OpPassManager& pm, llvm::StringRef module_name) {

  bool tf_mlir_enable_convert_control_to_data_outputs_pass;
  bool tf_mlir_enable_composite_tpuexecute_side_effects;
  bool tf_mlir_enable_strict_clusters;
  bool tf_mlir_enable_tpu_variable_runtime_reformatting_pass;
  // TODO(pineapplejuice233): Revisit this flag once the performance impact is verified
  // with different local CPU devices settings.
  bool tf_mlir_enable_multiple_local_cpu_devices;
};