本文主要是介绍tensorRT 自定义算子plugin的实现,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
文章目录
- 1. 自定义算子导出onnx
- 1.1 自定义算子的实现类
- 1.2 自定义算子类
- 1.3 onnx导出
- 2. 自定义算子plugin的实现
- 2.1 Plugin类的实现
- 2.1.1 构造函数的实现
- 2.1.2 析构函数实现
- 2.1.3 serialize函数的实现
- 2.1.4 destory函数的实现
- 2.1.5 enqueue函数的实现(关键)
- 2.1.5.1 customScalarImpl核函数的实现
- 2.1
这篇关于tensorRT 自定义算子plugin的实现的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!