REG_OP
函数原型
REG_OP(x)
功能说明
注册算子类型,同时自动生成算子类型的两个构造函数。
例如,注册算子的类型名称Conv2D,可调用REG_OP(Conv2D)接口,调用该接口后,定义了算子的类型名称Conv2D,同时产生Conv2D的两个构造函数,其中,Conv2D(const string& name)需指定算子名称,Conv2D()使用默认算子名称,例如“Conv2D唯一编号”。
class Conv2D : public Operator { typedef Conv2D _THIS_TYPE; public: explicit Conv2D(const string& name); explicit Conv2D(); }
参数说明
参数名 |
输入/输出 |
类型 |
描述 |
---|---|---|---|
x |
输入 |
- |
宏参数,被注册算子的类型名称 |
返回值
无。
异常处理
无。
约束说明
注册的算子类型名称需保持唯一,不能重复。