神经网络计算图优化方法
基本信息
| 申请号 | CN201810289195.4 | 申请日 | - |
| 公开（公告）号 | CN110321999B | 公开（公告）日 | 2021-10-01 |
| 申请公布号 | CN110321999B | 申请公布日 | 2021-10-01 |
| 分类号 | G06N3/04（2006.01）I;G06N3/08（2006.01）I | 分类 | 计算；推算；计数; |
| 发明人 | 隋凌志;王雨顺;刘鑫 | 申请（专利权）人 | 北京深鉴智能科技有限公司 |
| 代理机构 | 北京展翼知识产权代理事务所（特殊普通合伙） | 代理人 | 张阳 |
| 地址 | 100029 北京市朝阳区安定路5号院5号楼6层101号10层01-19单元 | ||
| 法律状态 | - | ||
摘要
| 摘要 | 本发明公开了一种神经网络计算图优化方法。所述计算图由计算平台用于执行神经网络计算，所述计算平台从外部存储器读取计算所需数据，所述方法包括：至少基于减少所述计算平台与所述外部存储器数据交互次数的优化规则选择能够进行层融合的层对象，根据选定的层对象融合所述计算图中至少两个相邻层，其中，所述至少两个相邻层是如下的至少一种：具有相同特征图输入的横向相邻层；以及前一层的特征图计算结果是后一层输入的至少一部分的竖向相邻层。本发明的计算图优化方案可以基于规则或通过同构子图匹配自动进行。由此，找出执行计算图的最优重构方式，提升神经网络计算平台的执行效率。 |
