更新时间:2021-03-18 GMT+08:00
GRAPH
函数功能
注册算子子图信息。
注册算子子图信息成功后,自动生成算子子图的相关接口,用户获取算子子图的名称、获取算子子图的描述、设置算子子图的描述。
例如,注册算子子图y,可调用OUTPUT(y)接口,注册算子子图成功后,自动生成以下相关接口:
static const string name_graph_y();// 返回输出的名称,即“y” SubgraphBuilder get_subgraph_builder_y() const;// 返回子图y对应的构建函数对象 _THIS_TYPE &set_subgraph_builder_y(const SubgraphBuilder &v);// 设置子图y对应的构建函数对象 Graph get_subgraph_y() const;// 设置子图y对应的graph对象
函数原型
OUTPUT (x)
参数说明
参数名 |
输入/输出 |
描述 |
---|---|---|
x |
输入 |
宏参数,算子子图的名称。 |
返回值
无。
异常处理
无。
约束说明
对于同一个算子,注册的算子子图名称需保持唯一,不能重复。
父主题: 算子原型注册宏
