ParseParamsFn函数
函数功能
参数解析函数。
函数原型
OpRegistrationData& ParseParamsFn(ParseParamFunc parseParamFn);
参数说明
参数 |
输入/输出 |
说明 |
---|---|---|
parseParamFn |
输入 |
回调函数ParseParamFunc,请参见回调函数ParseParamFunc。 |
回调函数ParseParamFunc
用户自定义并实现ParseParamFunc类函数,完成caffe模型参数和权值的转换,将结果填到Operator类中。
函数原型
Status ParseParamFunc(const Message* op_origin, ge::Operator& op_dest);
参数说明
参数 |
输入/输出 |
说明 |
---|---|---|
op_origin |
输入 |
protobuf格式的数据结构(来源于caffe模型的prototxt文件),包含算子参数信息。 |
op_dest |
输出 |
适配昇腾AI处理器的离线模型的算子数据结构,保存算子信息。 关于Operator类,请参见《GE API参考》中的“Operator类接口”。 |