def _save_function_alias(
    saved_model_dir: str,
    tags: Collection[str],
    function_aliases: Mapping[str, str],
) -> None:
  """Saves the function alias to the SavedModel.

  SavedModelBuilder (TF1 saved model saver) does not support saving function
  aliases, so this function loads the SavedModel proto and adds the
  `function_aliases` field.

  Args:

  ):
    # Convert any non-Tensor values from the sample to Tensors.
    # This conversion is required because the model saved in `model_dir` is
    # saved using TF1 SavedModelBuilder, which doesn't save the
    # SavedObjectGraph.
    func_kwargs = _convert_values_to_tf_tensors(sample)
    func(**func_kwargs)


def _run_graph_for_calibration_eager_mode(
    model_dir: str,