摘要: 如果模型中有些算子不被ONNX算子库支持,我们就需要利用ONNXRuntime提供的API手动添加新算子。在官方文档中已经对如何添加定制算子进行了介绍(https://onnxruntime.ai/docs/reference/operators/add-custom-op.html ),这里我们主 阅读全文
posted @ 2022-03-30 18:28 虔诚的树 阅读(5396) 评论(1) 推荐(1) 编辑
摘要: 背景:在学习如何往ONNXRuntime中添加新算子时,参考了官方测试代码:onnxruntime/test/shared_lib/test_inference.cc,代码内部使用GTest作为单元测试工具。为了清楚地学习运行过程,一步一步地调试是不可缺少的。 开始调试前需要以Debug方式编译代码 阅读全文
posted @ 2022-03-30 12:08 虔诚的树 阅读(1080) 评论(0) 推荐(0) 编辑