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

算子构建函数注册宏

宏功能

调用DOMI_REGISTER_OP完成算子注册。

宏原型

REGISTER_CUSTOM_OP(name)

参数说明

参数

输入/输出

说明

name

输入

算子在离线模型的类型名,可以随意指定,不可与已有类型名重复,区分大小写。

分享:

    相关文档

    相关产品