mlir::ModuleOp module_op, MLIRContext& ctx,
    absl::string_view calibration_data_dir,
    const bool force_regenerate_calibration_data) {
  // Disable DumpTensor ops when running calibration.
  DisableDebugging(module_op);

  std::vector<std::string> skipping_aggregator_ops;
  if (!force_regenerate_calibration_data) {
    TF_ASSIGN_OR_RETURN(const CalibrationStatisticsFlatMap statistics_map,

  absl::Status ExportToSavedModel(ModuleOp module_op,
                                  absl::string_view calibration_data_dir,
                                  bool force_regenerate_calibration_data,
                                  absl::string_view dst_saved_model_path);

  // Imports the SavedModel at `calibrated_saved_model_path` to `ModuleOp` after
  // running calibration.