code: def prepare_aot(aot: list[str], srcs_dir: str) -> None:
      """Rearrange xla_aot files in target the target directory.
  
      Args:
        aot: a list of paths to files that should be in xla_aot directory.
        srcs_dir: target directory where files are copied to.
      """
      for file in aot:
          if "external/local_tsl/" in file:
              copy_file(file, srcs_dir, "external/local_tsl/")
          elif "external/local_xla/" in file:
              copy_file(file, srcs_dir, "external/local_xla/")
          else:
              copy_file(file, srcs_dir)
  
      shutil.move(
          os.path.join(
              srcs_dir,
              (
                  "tensorflow/tools/pip_package/v2/"
                  "xla_build/CMakeLists.txt"
              ),
          ),
          os.path.join(srcs_dir, "CMakeLists.txt"),
      )
  
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 4 for calling a global function with example usage: prepare_aot(aot, srcs_dir) without return types