def postprocess_fn(
        batch: Dict[str, Any],  # pylint: disable=unused-argument
        auxiliary_output: Dict[str, Any]  # pylint: disable=unused-argument
    ) -> Dict[str, Any]:
      """Function that prepares model's input and output for serialization."""
      return {}
Function that prepares model's input and output for serialization.
language/mentionmemory/encoders/base_encoder.py