更新时间:2021-03-18 GMT+08:00
算子构建函数注册宏
宏功能
调用DOMI_REGISTER_OP完成算子注册。
宏原型
REGISTER_CUSTOM_OP(name)
参数说明
参数 |
输入/输出 |
说明 |
---|---|---|
name |
输入 |
算子在离线模型的类型名,可以随意指定,不可与已有类型名重复,区分大小写。 |
父主题: 算子插件开发函数参考