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.

  TF_RETURN_IF_ERROR(
      RunCalibrationPasses(*cloned_module_ref, *context, calibration_data_dir,
                           quantization_options.calibration_options()
                               .force_regenerate_calibration_data()));
  if (!IsCalibrationRequired(*cloned_module_ref)) return ExportedModel();

  absl::StatusOr<ExportedModel> exported_model = ModuleOpToExportedModel(

  // directory.
  string calibration_data_dir = 4;

  // Whether to reuse the existing calibration data in `calibration_data_dir`.
  // Default to False.
  bool force_regenerate_calibration_data = 5;
}

// Quantization configuration for StableHLO Quantizer. This is the primary
// message containing all configurable options.
// Next ID: 9
message QuantizationConfig {

      calib_opts.calibration_parameters.num_bins = 512

  if calib_opts.calibration_data_dir:
    save_model.create_empty_output_dir(
        calib_opts.calibration_data_dir,
        overwrite=calib_opts.force_regenerate_calibration_data,
    )


def _populate_quantization_options_default_values(
    quantization_options: _QuantizationOptions,
) -> None:
  """Populates default values for QuantizationOptions.