Function main

examples/vulkan/export.py:146–379

()

def main() -> None:  # noqa: C901
    logger = logging.getLogger("")
    logger.setLevel(logging.INFO)

    parser = argparse.ArgumentParser()
    parser.add_argument(
        "-m",
        "--model_name",
        required=True,
        help=f"provide a model name. Valid ones: {list(MODEL_NAME_TO_MODEL.keys())}",
    )

    parser.add_argument(
        "-fp16",
        "--force_fp16",
        action=argparse.BooleanOptionalAction,
        default=False,
        help="Force fp32 tensors to be converted to fp16 internally. Input/s outputs "
        "will be converted to/from fp32 when entering/exiting the delegate. Default is "
        "False",
    )

    parser.add_argument(
        "--small_texture_limits",
        action=argparse.BooleanOptionalAction,
        default=False,
        help="sets the default texture limit to be (2048, 2048, 2048) which is "
        "compatible with more devices (i.e. desktop/laptop GPUs) compared to the "
        "default (16384, 16384, 2048) which is more targeted for mobile GPUs. Default "
        "is False.",
    )

    parser.add_argument(
        "--skip_memory_planning",
        action=argparse.BooleanOptionalAction,
        default=False,
        help="Skips memory planning pass while lowering, which can be used for "
        "debugging. Default is False.",
    )

    parser.add_argument(
        "-s",
        "--strict",
        action=argparse.BooleanOptionalAction,
        default=True,
        help="whether to export with strict mode. Default is True",
    )

    parser.add_argument(
        "-d",
        "--dynamic",
        action=argparse.BooleanOptionalAction,
        default=False,
        help="Enable dynamic shape support. Default is False",
    )

    parser.add_argument(
        "-r",