本文主要是介绍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的实现的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！
