Move implementation files to _internal and create public API shims

Co-authored-by: gramalingam <10075881+gramalingam@users.noreply.github.com>

Author: Copilot

onnxscript/__init__.py

@@ -67,7 +67,7 @@
 import importlib.metadata
 
 from .backend.onnx_export import export2python as proto2python
-from .main import export_onnx_lib, graph, script
+from ._internal.main import export_onnx_lib, graph, script
 
 # isort: off
 from .onnx_opset import (
@@ -128,7 +128,7 @@
 
 from . import ir, optimizer, rewriter, version_converter
 from ._internal.utils import external_tensor
-from .values import OnnxFunction, TracedOnnxFunction
+from ._internal.values import OnnxFunction, TracedOnnxFunction
 
 # Set DEBUG to True to enable additional debug checks
 DEBUG: bool = False

onnxscript/_internal/autocast.py

@@ -13,7 +13,7 @@
 from onnxscript import ir, tensor
 
 if TYPE_CHECKING:
-    from onnxscript import converter
+    from onnxscript._internal import converter
 
 # Conversions from python values to ONNX are used by both the script converter as well
 # as the eager-mode runtime and both need to be consistent. The script converter converts

onnxscript/converter.py onnxscript/_internal/converter.pyonnxscript/converter.py renamed to onnxscript/_internal/converter.py

@@ -20,9 +20,16 @@
 import onnx_ir as ir
 
 import onnxscript
-from onnxscript import irbuilder, onnx_types, sourceinfo, values
-from onnxscript import type_annotation as ta
-from onnxscript._internal import analysis, ast_utils, autocast, param_manipulation
+from onnxscript import onnx_types, sourceinfo
+from onnxscript._internal import (
+    analysis,
+    ast_utils,
+    autocast,
+    irbuilder,
+    param_manipulation,
+    type_annotation as ta,
+    values,
+)
 
 logger = logging.getLogger("onnxscript")
 

onnxscript/converter_test.py onnxscript/_internal/converter_test.pyonnxscript/converter_test.py renamed to onnxscript/_internal/converter_test.py

@@ -23,7 +23,8 @@
 
 import onnxscript
 import onnxscript.testing
-from onnxscript import BOOL, FLOAT, INT64, converter, graph, script, tensor
+from onnxscript import BOOL, FLOAT, INT64, graph, script, tensor
+from onnxscript._internal import converter
 from onnxscript.onnx_opset import opset11 as op11
 from onnxscript.onnx_opset import opset15 as op
 from tests.common import onnx_script_test_case, testutils