申请/专利权人：北京大学

申请日：2023-04-24

公开（公告）日：2024-04-16

公开（公告）号：CN116629330B

主分类号：G06N3/063

分类号：G06N3/063;G06N3/0464;G06N3/048;G06N3/08

优先权：

专利状态码：有效-授权

法律状态：2024.04.16#授权;2023.09.08#实质审查的生效;2023.08.22#公开

摘要：本公开提供了一种算子检测方法、装置以及计算机设备，其中，该方法包括：获取神经网络模型的模型文件；基于所述模型文件确定神经网络模型中各网络层的结构描述信息；对每个所述网络层的结构描述信息进行遍历，得到每个所述网络层的算子信息，并基于每个所述网络层的算子信息生成目标算子列表；所述算子信息用于指示各网络层中算子的属性信息和算子之间的依赖关系；对所述目标算子列表进行检查，得到检查结果；通过所述检查结果确定能够通过所述神经网络模型的硬件设备执行相应计算操作的算子。

主权项：1.一种算子检测方法，其特征在于，包括：获取神经网络模型的模型文件；基于所述模型文件确定神经网络模型中各网络层的结构描述信息；对每个所述网络层的结构描述信息进行遍历，得到每个所述网络层的算子信息，并基于每个所述网络层的算子信息生成目标算子列表；所述算子信息用于指示各网络层中算子的属性信息和算子之间的依赖关系；对所述目标算子列表进行检查，得到检查结果；所述对所述目标算子列表进行检查，得到检查结果，包括：获取算子融合规则；基于所述目标算子列表中算子之间的依赖关系，确定满足所述算子融合规则的目标算子；按照所述算子融合规则对所述目标算子进行融合，得到融合算子；基于所述融合算子对所述目标算子列表进行检查，得到检查结果；所述基于所述融合算子对所述目标算子列表进行检查，得到检查结果，包括：在确定出所属于各个数据维度的目标算子均按照所述算子融合规则进行融合之后，将所述目标算子列表中的目标算子替换为所述融合算子，得到替换后目标算子列表；对所述替换后目标算子列表进行检查，得到所述检查结果；对目标算子列表中参与融合的目标算子，在判断出该目标算子在各个计算路径均参与了融合的情况下，从目标算子列表中删除该参与融合的目标算子，并在目标算子列表中保留融合算子；否则同时保留融合算子和参与融合的目标算子；通过所述检查结果确定能够通过所述神经网络模型的硬件设备执行相应计算操作的算子。

全文数据：

权利要求：

百度查询： 北京大学 一种算子检测方法、装置以及计算机设备

免责声明
1、本报告根据公开、合法渠道获得相关数据和信息，力求客观、公正，但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解，仅供参考使用，不能作为本公司承担任何法律责任的依据或者凭证。

阅读全文 双屏查看 官方信息 专利公告 收藏专利 下载PDF 下载WORD