[IR] Expose the convenience module to public

microsoft · Nov 19, 2024 · 66e3169 · 66e3169
1 parent db83343
commit 66e3169
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/onnxscript/ir/__init__.py b/onnxscript/ir/__init__.py
@@ -5,6 +5,7 @@
 __all__ = [
     # Modules
     "serde",
+    "convenience",
     # IR classes
     "Tensor",
     "ExternalTensor",
@@ -77,7 +78,7 @@
     "save",
 ]
 
-from onnxscript.ir import passes, serde, traversal
+from onnxscript.ir import convenience, passes, serde, traversal
 from onnxscript.ir._convenience import tensor
 from onnxscript.ir._core import (
     Attr,