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

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

输入

-

宏参数,被注册算子的类型名称

返回值

无。

异常处理

无。

约束说明

注册的算子类型名称需保持唯一,不能重复。

分享:

    相关文档

    相关产品