File size: 2,344 Bytes
6a22ec9 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 | # Copyright (c) Microsoft Corporation.
# Licensed under the MIT License.
__all__ = [
"script",
"graph",
"ir",
"optimizer",
"rewriter",
"version_converter",
"export_onnx_lib",
"OnnxFunction",
"TracedOnnxFunction",
"proto2python",
"external_tensor",
"BFLOAT16",
"FLOAT16",
"FLOAT8E4M3FN",
"FLOAT8E4M3FNUZ",
"FLOAT8E5M2",
"FLOAT8E5M2FNUZ",
"FLOAT",
"DOUBLE",
"INT8",
"INT16",
"INT32",
"INT64",
"UINT8",
"UINT16",
"UINT32",
"UINT64",
"BOOL",
"STRING",
"COMPLEX64",
"COMPLEX128",
"opset1",
"opset2",
"opset3",
"opset4",
"opset5",
"opset6",
"opset7",
"opset8",
"opset9",
"opset10",
"opset11",
"opset12",
"opset13",
"opset14",
"opset15",
"opset16",
"opset17",
"opset18",
"opset19",
"opset20",
"opset21",
"opset22",
"opset_ai_onnx_ml1",
"opset_ai_onnx_ml2",
"opset_ai_onnx_ml3",
"opset_ai_onnx_ml4",
"opset_ai_onnx_ml5",
"DEBUG",
]
import importlib.metadata
from .backend.onnx_export import export2python as proto2python
from .main import export_onnx_lib, graph, script
# isort: off
from .onnx_opset import (
opset1,
opset2,
opset3,
opset4,
opset5,
opset6,
opset7,
opset8,
opset9,
opset10,
opset11,
opset12,
opset13,
opset14,
opset15,
opset16,
opset17,
opset18,
opset19,
opset20,
opset21,
opset22,
opset_ai_onnx_ml1,
opset_ai_onnx_ml2,
opset_ai_onnx_ml3,
opset_ai_onnx_ml4,
opset_ai_onnx_ml5,
)
from .onnx_types import (
BFLOAT16,
FLOAT16,
FLOAT8E4M3FN,
FLOAT8E4M3FNUZ,
FLOAT8E5M2,
FLOAT8E5M2FNUZ,
FLOAT,
DOUBLE,
INT8,
INT16,
INT32,
INT64,
UINT8,
UINT16,
UINT32,
UINT64,
BOOL,
STRING,
COMPLEX64,
COMPLEX128,
)
# isort: on
from . import ir, optimizer, rewriter, version_converter
from ._internal.utils import external_tensor
from .values import OnnxFunction, TracedOnnxFunction
# Set DEBUG to True to enable additional debug checks
DEBUG: bool = False
try: # noqa: SIM105
__version__ = importlib.metadata.version("onnxscript")
except importlib.metadata.PackageNotFoundError:
# package is not installed
pass
|