microsoft/onnxruntime-extensions
Publicmirrored fromhttps://github.com/microsoft/onnxruntime-extensionsAvailable
onnxruntime_extensions/__init__.py
24lines · 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 | __version__ = "0.0.2" |
| 11 | __author__ = "Microsoft" |
| 12 | |
| 13 | |
| 14 | from ._ocos import get_library_path # noqa |
| 15 | from ._ocos import Opdef, PyCustomOpDef, hash_64, enable_custom_op # noqa |
| 16 | from ._ocos import expand_onnx_inputs # noqa |
| 17 | from ._ocos import hook_model_op # noqa |
| 18 | from ._ocos import default_opset_domain # noqa |
| 19 | from ._cuops import * # noqa |
| 20 | from .eager_op import EagerOp as PyOrtFunction |
| 21 | |
| 22 | |
| 23 | onnx_op = Opdef.declare |
| 24 | PyOp = PyCustomOpDef |
| 25 | |