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