onnxruntime/tools/python/util/__init__append.py

# appended to the __init__.py in the onnxruntime module's 'tools' folder from /tools/python/util/__init__append.py
import importlib.util
have_torch = importlib.util.find_spec("torch")
if have_torch:
    from .pytorch_export_helpers import infer_input_info  # noqa
Fix a couple of issues with the python package tools (#10858) * Tweaks to the model utils * Add handling for a dim_value of -1 when replacing the entire input shape. This occurs in models exported from PaddlePaddle * make pytorch helpers accessible in package * make QDQ helpers accessible in package 2022-03-15 05:52:12 +00:00			`# appended to the __init__.py in the onnxruntime module's 'tools' folder from /tools/python/util/__init__append.py`
			`import importlib.util`
			`have_torch = importlib.util.find_spec("torch")`
			`if have_torch:`
			`from .pytorch_export_helpers import infer_input_info # noqa`