Skip to content

Commit 74052ec

Browse files
authored
use the link from master branch (#206)
1 parent 493f51a commit 74052ec

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

PyTorchCustomOperator/README.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -200,7 +200,7 @@ struct GroupNormCustomOp : Ort::CustomOpBase<GroupNormCustomOp, GroupNormKernel<
200200
};
201201
```
202202

203-
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).
204204
Once you have the custom kernel and schema, you can add them to the domain using the C API as below:
205205
```cpp
206206
GroupNormCustomOp custom_op;
@@ -221,11 +221,11 @@ And include the required headers using ```include_directories```
221221
include_directories(<PATH_TO_EIGEN_HEADER_FILE>)
222222
```
223223

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).
225225

226226
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.
227227

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).
229229

230230

231231

0 commit comments

Comments
 (0)