Skip to content

Commit 029441f

Browse files
gramalingamCopilot
andauthored
Expose build_function and build_graph in public API (#2913)
Add build_function and build_graph to onnxscript's top-level exports, allowing downstream packages (e.g. mobius) to import them directly as onnxscript.build_function / onnxscript.build_graph instead of reaching into onnxscript._internal.builder. Signed-off-by: G Ramalingam <grama@microsoft.com> Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
1 parent e8d744f commit 029441f

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

onnxscript/__init__.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,8 @@
1616
"OpBuilder",
1717
"OpBuilderBase",
1818
"TapeBuilder",
19+
"build_function",
20+
"build_graph",
1921
"proto2python",
2022
"external_tensor",
2123
"BFLOAT16",
@@ -132,7 +134,7 @@
132134
# isort: on
133135

134136
from . import ir, nn, optimizer, rewriter, version_converter
135-
from ._internal.builder import GraphBuilder, OpBuilder
137+
from ._internal.builder import GraphBuilder, OpBuilder, build_function, build_graph
136138
from ._internal.tape_builder import OpBuilderBase, TapeBuilder
137139
from ._internal.utils import external_tensor
138140
from ._internal.values import OnnxFunction, TracedOnnxFunction

0 commit comments

Comments
 (0)