[发明专利]一种神经网络的内存分配方法及装置有效
| 申请号： | 201810607693.9 | 申请日： | 2018-06-13 |
| 公开（公告）号： | CN110597616B | 公开（公告）日： | 2022-07-29 |
| 发明（设计）人： | 黄莉华;丁皓;范翱翔 | 申请（专利权）人： | 华为技术有限公司 |
| 主分类号： | G06F9/50 | 分类号： | G06F9/50;G06N3/063 |
| 代理公司： | 北京同达信恒知识产权代理有限公司 11291 | 代理人： | 冯艳莲 |
| 地址： | 518129 广东*** | 国省代码： | 广东;44 |
| 权利要求书： | 查看更多 | 说明书： | 查看更多 |
| 摘要： | 本申请涉及人工智能或者机器学习领域，具体提供一种神经网络的内存分配方法及装置，神经网络在运行时，需占用N个内存块，该方法包括：确定N个内存块的信息；根据N个内存块的大小，对N个内存块进行由大到小排序，获得N个内存块的排序序列；分配第一内存块给神经网络，用于进行神经网络中第一节点集合的运算；判断第n内存块是否可复用已分配集合中的内存块，已分配集合中包括已经分配给神经网络的内存块，如果第n内存块可复用已分配集合中的内存块，则将可复用的内存块分配给神经网络，用于进行神经网络中第n节点集合的运算，并更新已分配集合；其中，所述第n内存块为第二内存块至第N内存块中的任一内存块。采用本申请的方法及装置，可优化神经网络的内存分配。 | ||
| 搜索关键词： | 一种 神经网络 内存 分配 方法 装置 | ||
【主权项】：
1.一种神经网络的内存分配方法，其特征在于，所述神经网络在运行时，需占用N个内存块，所述N为大于等于1的整数，所述方法包括：/n确定所述N个内存块的信息，所述N个内存块的信息包括所述N个内存块的大小和所述N个内存块中每个内存块一一对应的节点集合，所述每个内存块用于进行所述神经网络中与所述每个内存块一一对应的节点集合的运算；/n根据所述N个内存块的大小，对所述N个内存块进行由大到小排序，获得所述N个内存块的排序序列；/n分配第一内存块给所述神经网络，用于进行所述神经网络中第一节点集合的运算，其中，所述第一内存块为所述N个内存块中大小最大的内存块；/n判断第n内存块是否可复用已分配集合中的内存块，所述已分配集合中包括已经分配给所述神经网络的内存块，如果所述第n内存块可复用所述已分配集合中的内存块，则将所述可复用的内存块分配给所述神经网络，用于进行所述神经网络中第n节点集合的运算，并更新所述已分配集合，其中，所述n为大于或等于2，小于或等于N的正整数；/n根据所述排序序列，从n＝2到n＝N依次执行上述判断，若判断为是，则更新所述已分配集合。/n
下载完整专利技术内容需要扣除积分，VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息，商用须获得专利权人授权。该专利全部权利属于华为技术有限公司，未经华为技术有限公司许可，擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作，请联系【客服】
本文链接：http://www.vipzhuanli.com/patent/201810607693.9/，转载请声明来源钻瓜专利网。
