exir/__init__.py

# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

from typing import Any

from executorch.exir.capture import (
    _capture_legacy_do_not_use,
    CallSpec,
    capture,
    CaptureConfig,
    EdgeCompileConfig,
    ExecutorchBackendConfig,
)
from executorch.exir.emit import emit_program, EmitterOutput
from executorch.exir.program import (
    _to_edge,
    edge_to_executorch_passes,
    EdgeProgramManager,
    ExecutorchProgram,
    ExecutorchProgramManager,
    ExirExportedProgram,
    to_edge,
    to_edge_transform_and_lower,
)
from executorch.exir.serde.serialize import load, save
from executorch.exir.tracer import ExirDynamoConfig
from torch.export import ExportedProgram, ExportGraphSignature

Value = Any

__all__ = [
    "emit_program",
    "EmitterOutput",
    "capture",
    "_capture_legacy_do_not_use",
    "CallSpec",
    "ExportedProgram",
    "ExirExportedProgram",
    "ExecutorchProgram",
    "ExportGraphSignature",
    "_to_edge",
    "to_edge",
    "to_edge_transform_and_lower",
    "EdgeProgramManager",
    "ExecutorchProgramManager",
    "edge_to_executorch_passes",
    "CaptureConfig",
    "EdgeCompileConfig",
    "ExecutorchBackendConfig",
    "Value",
    "ExirDynamoConfig",
    "load",
    "save",
]