microsoft/onnxruntime-extensions
Publicmirrored fromhttps://github.com/microsoft/onnxruntime-extensionsAvailable
onnxruntime_extensions/__init__.py
27lines · modecode
| 1 | # Copyright (c) Microsoft Corporation. All rights reserved. |
| 2 | # Licensed under the MIT License. See License.txt in the project root for |
| 3 | # license information. |
| 4 | ############################################################################### |
| 5 | |
| 6 | """ |
| 7 | The entry point to onnxruntime custom op library |
| 8 | """ |
| 9 | |
| 10 | __author__ = "Microsoft" |
| 11 | |
| 12 | from ._version import __version__ |
| 13 | from ._ocos import get_library_path # noqa |
| 14 | from ._ocos import Opdef, PyCustomOpDef # noqa |
| 15 | from ._ocos import hash_64 # noqa |
| 16 | from ._ocos import enable_py_op # noqa |
| 17 | from ._ocos import expand_onnx_inputs # noqa |
| 18 | from ._ocos import hook_model_op # noqa |
| 19 | from ._ocos import default_opset_domain # noqa |
| 20 | from ._cuops import * # noqa |
| 21 | from ._ortapi2 import OrtPyFunction as PyOrtFunction # backward compatibility |
| 22 | from ._ortapi2 import OrtPyFunction, optimize_model, make_onnx_model, ONNXRuntimeError |
| 23 | |
| 24 | |
| 25 | onnx_op = Opdef.declare |
| 26 | PyOp = PyCustomOpDef |
| 27 | |
| 28 | |