return graph_def


def create_empty_output_dir(
    output_directory: str, overwrite: bool = True
) -> None:
  """Creates the `output_directory`.

  If `output_directory` already exists, it recursively deletes all contents
  inside the directory.

  Also creates the parent & intermediate directories.

  Args:
    output_directory: Output directory.
    overwrite: Where to clean the output directory if exists.