code: def prepare_wheel_srcs(
      headers: list[str],
      srcs: list[str],
      aot: list[str],
      srcs_dir: str,
      version: str,
  ) -> None:
      """Rearrange source and header files.
  
      Args:
        headers: a list of paths to header files.
        srcs: a list of paths to the rest of files.
        aot: a list of paths to files that should be in xla_aot directory.
        srcs_dir: directory to copy files to.
        version: tensorflow version.
      """
      prepare_headers(
          headers, os.path.join(srcs_dir, "tensorflow/include")
      )
      prepare_srcs(srcs, srcs_dir)
      prepare_aot(
          aot, os.path.join(srcs_dir, "tensorflow/xla_aot_runtime_src")
      )
  
      # Every directory that contains a .py file gets an empty __init__.py file.
      create_init_files(os.path.join(srcs_dir, "tensorflow"))
  
      # move MANIFEST and THIRD_PARTY_NOTICES to the root
      shutil.move(
          os.path.join(
              srcs_dir, "tensorflow/tools/pip_package/v2/MANIFEST.in"
          ),
          os.path.join(srcs_dir, "MANIFEST.in"),
      )
      shutil.move(
          os.path.join(
              srcs_dir,
              "tensorflow/tools/pip_package/THIRD_PARTY_NOTICES.txt",
          ),
          os.path.join(srcs_dir, "tensorflow/THIRD_PARTY_NOTICES.txt"),
      )
  
      update_xla_tsl_imports(os.path.join(srcs_dir, "tensorflow"))
      if not is_windows():
          rename_libtensorflow(
              os.path.join(srcs_dir, "tensorflow"), version
          )
      if not is_macos() and not is_windows():
          patch_so(srcs_dir)
  
imports: import argparse
  import glob
  import os
  import shutil
  import subprocess
  import sys
  import tempfile
  from tensorflow.tools.pip_package.v2.utils.utils import copy_file
  from tensorflow.tools.pip_package.v2.utils.utils import create_init_files
  from tensorflow.tools.pip_package.v2.utils.utils import is_macos
  from tensorflow.tools.pip_package.v2.utils.utils import is_windows
  from tensorflow.tools.pip_package.v2.utils.utils import replace_inplace
label_desc: Use image node_id 5 for calling a global function with example usage: prepare_wheel_srcs(headers, srcs, aot, srcs_dir, version) without return types