更新时间:2021-03-18 GMT+08:00
分享

开发流程总览

TE自定义算子开发流程如图1所示。

图1 TE自定义算子开发流程

主要流程如下:

  1. 创建自定义算子开发工程。
  2. 自定义未实现的算子,包括算子代码开发、单算子的编译、运行、验证。
  3. 自定义算子插件开发,将算子注册到Framework中,算子插件编译后,生成自定义算子的*.so插件文件。
  4. 再次进行模型转换时,通过加载*.so插件文件,才能识别自定义的算子,进而完成模型转换。
分享:

    相关文档

    相关产品