def _run_graph_for_calibration(
    float_model_dir: str,
    signature_keys: Sequence[str],
    tags: Collection[str],
    representative_dataset: rd.RepresentativeDatasetOrMapping,
    force_graph_mode_calibration: bool,
) -> None:
  """Runs the graph for calibration using representative datasets.

  Args:
    float_model_dir: Path to the model to calibrate.

    )

  # Set `force_graph_mode_calibration` to True to avoid skipping op execution,
  # which are not connected to return ops, during calibration execution.
  # TODO: b/335031954 - Bring back support to run calibration in Eager mode.
  logging.debug(
      'Setting `force_graph_mode_calibration = True` to ensure the calibration'
      ' mode is executed properly.'
  )

        saved_model_path, tags, signature_def_map);

    py_function_library.RunCalibration(
        calibration_saved_model_dir, signature_keys, tags,
        quantization_options.force_graph_mode_calibration(),
        representative_dataset_file_map_serialized);
  }

  if (absl::Status status = AddCalibrationStatistics(
          *module_ref, calibration_data_dir,