xref: /aosp_15_r20/external/executorch/exir/__init__.py (revision 523fa7a60841cd1ecfb9cc4201f1ca8b03ed023a) 
1# Copyright (c) Meta Platforms, Inc. and affiliates.
2# All rights reserved.
3#
4# This source code is licensed under the BSD-style license found in the
5# LICENSE file in the root directory of this source tree.
6
7from typing import Any
8
9from executorch.exir.capture import (
10    _capture_legacy_do_not_use,
11    CallSpec,
12    capture,
13    CaptureConfig,
14    EdgeCompileConfig,
15    ExecutorchBackendConfig,
16)
17from executorch.exir.emit import emit_program, EmitterOutput
18from executorch.exir.program import (
19    _to_edge,
20    edge_to_executorch_passes,
21    EdgeProgramManager,
22    ExecutorchProgram,
23    ExecutorchProgramManager,
24    ExirExportedProgram,
25    to_edge,
26    to_edge_transform_and_lower,
27)
28from executorch.exir.serde.serialize import load, save
29from executorch.exir.tracer import ExirDynamoConfig
30from torch.export import ExportedProgram, ExportGraphSignature
31
32Value = Any
33
34__all__ = [
35    "emit_program",
36    "EmitterOutput",
37    "capture",
38    "_capture_legacy_do_not_use",
39    "CallSpec",
40    "ExportedProgram",
41    "ExirExportedProgram",
42    "ExecutorchProgram",
43    "ExportGraphSignature",
44    "_to_edge",
45    "to_edge",
46    "to_edge_transform_and_lower",
47    "EdgeProgramManager",
48    "ExecutorchProgramManager",
49    "edge_to_executorch_passes",
50    "CaptureConfig",
51    "EdgeCompileConfig",
52    "ExecutorchBackendConfig",
53    "Value",
54    "ExirDynamoConfig",
55    "load",
56    "save",
57]
58