注册算子输出,调用该接口后会返回一个OpParamDef结构,后续可通过该结构配置算子输出信息。
1
OpParamDef &Output(const char *name)
参数
输入/输出
说明
name
输入
算子输出名称。
算子参数定义,OpParamDef实例,具体请参考OpParamDef。
参数注册的顺序需要和算子kernel入口函数一致。