更新时间:2021-03-18 GMT+08:00
分享

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类接口”

分享:

    相关文档

    相关产品