You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
The Compute function is implemented [in the source file](https://github.com/neginraoof/CustomOperators/blob/master/CuctomOperator/ort_custom_op/custom_op.cc).
203
+
The Compute function is implemented [in the source file](https://github.com/onnx/tutorials/tree/master/PyTorchCustomOperator/ort_custom_op/custom_op.cc).
204
204
Once you have the custom kernel and schema, you can add them to the domain using the C API as below:
205
205
```cpp
206
206
GroupNormCustomOp custom_op;
@@ -221,11 +221,11 @@ And include the required headers using ```include_directories```
221
221
include_directories(<PATH_TO_EIGEN_HEADER_FILE>)
222
222
```
223
223
224
-
An example ```CMakeLists.txt``` file we could be found [here](https://github.com/neginraoof/CustomOperators/blob/master/CuctomOperator/ort_custom_op/CMakeLists.txt).
224
+
An example ```CMakeLists.txt``` file we could be found [here](https://github.com/onnx/tutorials/tree/master/PyTorchCustomOperator/ort_custom_op/CMakeLists.txt).
225
225
226
226
Once you have the cmake file, create a build directory from the same location and try ```cd build```. Execute the command ```cmake ..``` to configure the project and build it using ```make``` command.
227
227
228
-
Now that you have registered your operator, you should be able to run your model and test it. You can find the source code and test for a sample custom operator [here](https://github.com/neginraoof/CustomOperators/blob/master/CuctomOperator/ort_custom_op/custom_op_test.cc).
228
+
Now that you have registered your operator, you should be able to run your model and test it. You can find the source code and test for a sample custom operator [here](https://github.com/onnx/tutorials/tree/master/PyTorchCustomOperator/ort_custom_op/custom_op_test.cc).
0 commit comments