microsoft/onnxruntime-extensions
Publicmirrored fromhttps://github.com/microsoft/onnxruntime-extensionsAvailable
onnxruntime_extensions/onnxprocess/__init__.py
12lines · modecode
| 1 | """ |
| 2 | override the torch importing, to dump all torch operators during the processing code. |
| 3 | !!!This package depends on onnxruntime_extensions root package, but not vice versa.!!! |
| 4 | , since this package fully relies on pytorch, while the onnxruntime_extensions doesn't |
| 5 | """ |
| 6 | |
| 7 | from . _tensor import op_from_customop as pyfunc_from_custom_op |
| 8 | from . _tensor import op_from_model as pyfunc_from_model |
| 9 | from ._builder import build_customop_model |
| 10 | from ._session import ONNXTraceSession |
| 11 | |
| 12 | trace_for_onnx = ONNXTraceSession.trace_for_onnx |
| 13 | |