更新时间:2021-10-26 GMT+08:00
算子类型注册宏
宏功能
对一种算子类型进行注册,主要是声明一个常量字符串,该字符串就是算子类型的名称,然后将其保存到一个set中。
宏原型
REGISTER_OPTYPE_DECLARE(var_name, str_name)
REGISTER_OPTYPE_DEFINE(var_name, str_name)
参数说明
参数 |
输入/输出 |
说明 |
---|---|---|
var_name |
输入 |
常量字符串名称 |
str_name |
输入 |
算子类型名称 |
父主题: 宏定义