You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
10 lines
512 B
10 lines
512 B
# -------------------------------------------------------------------------
|
|
# 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
|