__init__.py 518 B

12345678910
  1. # -------------------------------------------------------------------------
  2. # Copyright (c) Microsoft Corporation. All rights reserved.
  3. # Licensed under the MIT License.
  4. # --------------------------------------------------------------------------
  5. # appended to the __init__.py in the onnxruntime module's 'tools' folder from /tools/python/util/__init__append.py
  6. import importlib.util
  7. have_torch = importlib.util.find_spec("torch")
  8. if have_torch:
  9. from .pytorch_export_helpers import infer_input_info # noqa: F401