11 lines
518 B
Python
11 lines
518 B
Python
# -------------------------------------------------------------------------
|
|
# Copyright (c) Microsoft Corporation. All rights reserved.
|
|
# Licensed under the MIT License.
|
|
# --------------------------------------------------------------------------
|
|
# 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: F401
|