// and is used for debugging. Changing the `name` does not modify the behavior
// of the passes.
//
// It will try to dump intermediate MLIRs if certain conditions are met. See the
// description from `MaybeEnableIrPrinting` for the details about the
// conditions.
//
// Returns a non-OK status when the pass run fails or it fails to create an MLIR
// dump file.
template <typename FuncT>

                                                    /*propagate=*/true);

  absl::StatusOr<std::unique_ptr<llvm::raw_ostream>> dump_file;
  if (mlir_dump_file_name) {
    TF_RETURN_IF_ERROR(tensorflow::quantization::MaybeEnableIrPrinting(
        pass_manager, mlir_dump_file_name.value()));
  }

  if (failed(pass_manager.run(module_op))) {
    return statusHandler.ConsumeStatus();
  }

  return absl::OkStatus();
}

//
// `mlir_dump_file_prefix` is primarily used for debugging and does not affect
// the preprocessing behavior. Instructions for producing MLIR dump files are in
// the comments of `tensorflow::quantization::MaybeEnableIrPrinting` function.
absl::Status PreprocessAndFreezeGraph(
    absl::string_view mlir_dump_file_prefix, bool is_inliner_run,
    const absl::flat_hash_set<std::string>& noinline_functions,

// * `pass_number` increments from 1 for each pass.
// * `pass_name` is the name of the pass.
// * `before|after` indicates whether the dump occurs before or after the pass.
absl::Status MaybeEnableIrPrinting(mlir::PassManager &pm,
                                   absl::string_view file_name_prefix);

}  // namespace quantization
}  // namespace tensorflow

  pm.enableIRPrinting(std::make_unique<PrinterConfig>(
      file_name_prefix, /*print_module_scope=*/false,
      /*print_after_only_on_change=*/true, flag));
}

absl::Status MaybeEnableIrPrinting(mlir::PassManager& pm,
                                   absl::string_view file_name_prefix) {
  if (!VLOG_IS_ON(1)) {
    LOG(INFO) << "Verbosity level too low to enable IR printing.";
    return absl::OkStatus();