[发明专利]神经网络模型的编译方法和装置、电子设备和存储介质在审
申请号: | 202111519387.8 | 申请日: | 2021-12-13 |
公开(公告)号: | CN114239803A | 公开(公告)日: | 2022-03-25 |
发明(设计)人: | 李德林;张祎男;李建军;王振江 | 申请(专利权)人: | 北京地平线机器人技术研发有限公司 |
主分类号: | G06N3/04 | 分类号: | G06N3/04;G06N3/063;G06N3/08 |
代理公司: | 北京思源智汇知识产权代理有限公司 11657 | 代理人: | 毛丽琴 |
地址: | 100094 北京市海淀*** | 国省代码: | 北京;11 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | 本公开实施例公开了一种神经网络模型的编译方法和装置、电子设备和存储介质,其中,方法包括:获取待编译神经网络模型对应的第一模型;针对第一模型的至少一个MUL层中的第一MUL层,将第一MUL层的第一输入张量和第二输入张量的元素对应相乘运算变换为深度卷积运算,并将深度卷积运算获得的第一输出张量变换为与第一MUL层的原输出张量数据结构相同的第二输出张量,获得第一MUL层对应的优化后的第二MUL层;基于各优化后的第二MUL层获得第一模型对应的优化模型;将优化模型编译为对应的二进制的目标模型。本公开实施例使得神经网络模型的元素对应相乘运算可以在加速芯片上实现,从而有效提高神经网络模型的推理速度。 | ||
搜索关键词: | 神经网络 模型 编译 方法 装置 电子设备 存储 介质 | ||
【主权项】:
暂无信息
下载完整专利技术内容需要扣除积分,VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京地平线机器人技术研发有限公司,未经北京地平线机器人技术研发有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/patent/202111519387.8/,转载请声明来源钻瓜专利网。