mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-15 20:50:42 +00:00
f385c73058
* 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
5 lines
264 B
Python
5 lines
264 B
Python
# 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
|