更新时间:2021-03-18 GMT+08:00
开发流程总览
TE自定义算子开发流程如图1所示。
主要流程如下:
- 创建自定义算子开发工程。
- 自定义未实现的算子,包括算子代码开发、单算子的编译、运行、验证。
- 自定义算子插件开发,将算子注册到Framework中,算子插件编译后,生成自定义算子的*.so插件文件。
- 再次进行模型转换时,通过加载*.so插件文件,才能识别自定义的算子,进而完成模型转换。