mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-15 20:50:42 +00:00
af9c554dd3
* Prototype version that demonstrates it can work * Switched to OrtValue and removed the OrtCustomOpTensor code. * Support multiple outputs and reading of attributes * Add custom domain handling to custom ops * Update documentation * more wording changes
23 lines
1.4 KiB
Markdown
23 lines
1.4 KiB
Markdown
Adding a new op
|
|
===============
|
|
|
|
## A new op can be written and registered with ONNXRuntime in the following 3 ways
|
|
### 1. Using the experimental custom op API in the C API (onnxruntime_c_api.h)
|
|
Note: These APIs are experimental and will change in the next release. They're released now for feedback and experimentation.
|
|
* Create an OrtCustomOpDomain with the domain name used by the custom ops
|
|
* Create an OrtCustomOp structure for each op and add them to the OrtCustomOpDomain with OrtCustomOpDomain_Add
|
|
* Call OrtAddCustomOpDomain to add the custom domain of ops to the session options
|
|
See [this](../onnxruntime/test/custom_op_shared_lib/test_custom_op.cc) for an example.
|
|
|
|
### 2. Using RegisterCustomRegistry API
|
|
* Implement your kernel and schema (if required) using the OpKernel and OpSchema APIs (headers are in the include folder).
|
|
* Create a CustomRegistry object and register your kernel and schema with this registry.
|
|
* Register the custom registry with ONNXRuntime using RegisterCustomRegistry API.
|
|
|
|
See
|
|
[this](../onnxruntime/test/framework/local_kernel_registry_test.cc) for an example.
|
|
|
|
### 3. Contributing the op to ONNXRuntime
|
|
This is mostly meant for ops that are in the process of being proposed to ONNX. This way you don't have to wait for an approval from the ONNX team
|
|
if the op is required in production today.
|
|
See [this](../onnxruntime/contrib_ops) for an example.
|