更新时间: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

输入

宏参数,算子子图的名称。

返回值

无。

异常处理

无。

约束说明

对于同一个算子,注册的算子子图名称需保持唯一,不能重复。

分享:

    相关文档

    相关产品

close