[发明专利]一种用于神经网络的算子融合方法及其相关产品有效
申请号: | 201910750140.3 | 申请日: | 2019-08-14 |
公开(公告)号: | CN110490309B | 公开(公告)日: | 2022-06-07 |
发明(设计)人: | 不公告发明人 | 申请(专利权)人: | 中科寒武纪科技股份有限公司 |
主分类号: | G06N3/06 | 分类号: | G06N3/06;G06N3/04 |
代理公司: | 北京维昊知识产权代理事务所(普通合伙) 11804 | 代理人: | 李波;孙新国 |
地址: | 100086 北京市海*** | 国省代码: | 北京;11 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | 本公开涉及一种可用于神经网络的算子融合方法及相关产品,其中,所述算子融合方法包括:遍历所述神经网络对应的计算图中的算子,从所述计算图中选择当前算子和待融合算子;按照一拆分尺寸,将所述待融合算子的输出数据拆分成多个输出块;基于所述输出块,映射得到与所述输出块对应的所述当前算子的输入数据的数据块尺寸、以及所述当前算子与待融合算子之间的中间数据的数据块尺寸;基于所述拆分尺寸、所述当前算子的输入数据的数据块尺寸、以及所述当前算子与待融合算子之间的中间数据的数据块尺寸,判断是否将所述当前算子与待融合算子融合在一起。 | ||
搜索关键词: | 一种 用于 神经网络 算子 融合 方法 及其 相关 产品 | ||
【主权项】:
1.一种可用于神经网络的算子融合方法,其特征在于,所述算子融合方法包括:/n遍历所述神经网络对应的计算图中的算子,从所述计算图中选择当前算子和待融合算子;/n按照一拆分尺寸,将所述待融合算子的输出数据拆分成多个输出块;/n基于所述输出块,映射得到与所述输出块对应的所述当前算子的输入数据的数据块尺寸、以及所述当前算子与待融合算子之间的中间数据的数据块尺寸;/n基于所述拆分尺寸、所述当前算子的输入数据的数据块尺寸、以及所述当前算子与待融合算子之间的中间数据的数据块尺寸,判断是否将所述当前算子与待融合算子融合在一起。/n
下载完整专利技术内容需要扣除积分,VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于中科寒武纪科技股份有限公司,未经中科寒武纪科技股份有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/patent/201910750140.3/,转载请声明来源钻瓜专利网。