[发明专利]一种模型算子对比方法、系统及存储介质

权利要求书

1.一种模型算子对比方法，其特征在于，包括如下步骤：

获取源模型，所述源模型包括路径和网络结构定义文件；

判断源模型类型，得到所述源模型类型；

基于所述源模型类型、所述路径和所述网络结构定义文件，解析得到所述源模型的第一算子；

获取目标模型，读取支持转换到所述目标模型的第二算子；将所述第二算子与所述第一算子进行对比，若所述第一算子中存在第三算子，所述第三算子不包含于所述第二算子，则提示所述第三算子不支持转换到所述目标模型；若不存在所述第三算子，则提示支持所述源模型转换到所述目标模型。

2.根据权利要求1所述的模型算子对比方法，其特征在于，所述判断源模型类型，得到所述源模型类型步骤进一步包括：

获取所述源模型的模型后缀，根据所述模型后缀判断源模型类型；

根据所述源模型类型执行模型加载的操作；如果加载成功，则输出所述源模型类型；如果加载失败，则提示所述源模型类型判断错误。

3.根据权利要求2所述的模型算子对比方法，其特征在于，每个所述源模型类型具有若干所述模型后缀。

4.根据权利要求1所述的模型算子对比方法，其特征在于，所述解析得到所述源模型的第一算子步骤进一步包括：

给所述源模型设置输入，使所述源模型进行前向传播，得到计算图；

遍历所述计算图，得到并输出所述第一算子。

5.根据权利要求4所述的模型算子对比方法，其特征在于，所述遍历所述计算图步骤进一步包括：

通过反向传播遍历所述计算图。

6.根据权利要求1所述的模型算子对比方法，其特征在于，所述读取支持转换到所述目标模型的第二算子的步骤进一步包括：

将支持转换到所述目标模型的第二算子按类型分别进行存储。

7.根据权利要求5所述的模型算子对比方法，其特征在于：对支持转换到所述目标模型的所述第二算子按类型分别进行更新。

8.一种模型算子对比系统，其特征在于，包括：类型判断模块、模型解析模块和算子对比模块；

所述类型判断模块用于获取源模型和模型后缀，根据所述模型后缀和模型加载操作判断输入模型的类型，并将得到的模型类型输出到所述模型解析模块；

所述模型解析模块用于接收所述模型类型，根据所述模型类型对所述输入模型进行解析，得到所述输入模型的第一算子，并将所述第一算子输出到所述算子对比模块；

所述算子对比模块用于接收所述第一算子；获取目标模型，根据所述目标模型类型读取支持转换到所述目标模型的第二算子，将所述第一算子与读取的所述第二算子进行对比，得到所述第一算子中不支持转换到所述目标模型的第三算子。

9.根据权利要求8所述的模型算子对比系统，其特征在于，所述系统还包括更新维护模块；

所述更新维护模块用于对支持转换到所述目标模型的所述第二算子进行更新和维护，输出新增加的所述第二算子以及删除的所述第二算子作为提示。

10.一种计算机存储介质，其特征在于，用于储存为上述权利要求1－7中任一项所述的模型算子对比方法所用的计算机软件指令，其包含用于执行上述为模型算子对比方法所设计的程序。