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

插件实现流程

自定义算子开发完成后,离线模型生成器需要支持将自定义算子的属性值转换到离线模型,在离线模型执行阶段能够支持自定义算子的执行,所以用户需要开发自定义算子插件,进行算子属性的解析、形状类别的推断,并通过离线模型生成器提供的注册机制完成自定义算子的注册。

图1 自定义算子插件实现流程图
分享:

    相关文档

    相关产品