Output

函数功能

注册算子输出,调用该接口后会返回一个OpParamDef结构,后续可通过该结构配置算子输出信息。

函数原型

1
OpParamDef &Output(const char *name)

参数说明

参数

输入/输出

说明

name

输入

算子输出名称。

返回值说明

算子参数定义,OpParamDef实例,具体请参考OpParamDef

约束说明

参数注册的顺序需要和算子kernel入口函数一致。