[发明专利]一种神经网络剪枝方法、装置、计算机设备及存储介质有效
申请号: | 201910914712.7 | 申请日: | 2019-09-24 |
公开(公告)号: | CN110689116B | 公开(公告)日: | 2022-12-27 |
发明(设计)人: | 不公告发明人 | 申请(专利权)人: | 安徽寒武纪信息科技有限公司 |
主分类号: | G06N3/04 | 分类号: | G06N3/04;G06N3/08 |
代理公司: | 广州三环专利商标代理有限公司 44202 | 代理人: | 郝传鑫;熊永强 |
地址: | 231283 安徽省合肥市高新区习友路3333*** | 国省代码: | 安徽;34 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | 本申请实施例公开了一种神经网络剪枝方法、装置、计算机设备及存储介质,可以对运行在人工智能处理器上的算子对应的计算图进行剪枝,提高异构系统并行执行神经网络计算任务的效率。 | ||
搜索关键词: | 一种 神经网络 剪枝 方法 装置 计算机 设备 存储 介质 | ||
【主权项】:
1.一种神经网络剪枝方法,其特征在于,所述方法包括:/n获取神经网络模型对应的原始计算图,从所述原始计算图中确定第一类算子;其中,所述第一类算子对应的操作指令能够运行在人工智能处理器上;/n根据所述原始计算图中算子之间具有的有向边,获取所述第一类算子构成的计算图,在所述第一类算子构成的计算图中提取原始子图;其中,所述原始子图包含多个输入算子和/或多个输出算子;所述原始子图构成原始子图集合;/n对所述原始子图进行剪枝,得到对应的剪枝结果子图;其中,所述剪枝结果子图包含一个输入算子和一个输出算子;/n获取所述剪枝结果子图对应的操作指令,以分配至对应的人工智能处理器上执行任务。/n
下载完整专利技术内容需要扣除积分,VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于安徽寒武纪信息科技有限公司,未经安徽寒武纪信息科技有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/patent/201910914712.7/,转载请声明来源钻瓜专利网。