[发明专利]一种应用于AI异构编译器的自动切分神经网络子图方法有效
申请号: | 201911312625.0 | 申请日: | 2019-12-18 |
公开(公告)号: | CN111062467B | 公开(公告)日: | 2023-05-12 |
发明(设计)人: | 黄明飞;王海涛;吕春莹 | 申请(专利权)人: | 开放智能机器(上海)有限公司 |
主分类号: | G06N3/0464 | 分类号: | G06N3/0464;G06N3/063;G06N3/08 |
代理公司: | 上海申新律师事务所 31272 | 代理人: | 党蕾 |
地址: | 200233 上海市徐汇区*** | 国省代码: | 上海;31 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | 本发明公开一种应用于AI异构编译器的自动切分神经网络子图方法,其中包括,创建节点列表;预先对神经网络计算图的所有节点进行编号,并记录每个节点的输入依赖关系与输出依赖关系;从神经网络计算图中获取任一输入无依赖关系的节点;删除输入无依赖关系的节点以及以输入无依赖关系的节点为起点的所有连接边,且将删除的节点的编号存储至节点列表中;重复步骤S2与步骤S3,直至神经网络计算图为空,将删除的节点的编号存储至节点列表中;根据节点列表中的节点属性颜色进行自动切分,以处理得到各个神经网络计算子图。有益效果:适用于任一复杂结构的神经网络计算图,解决了神经网络计算图不好切分子图以及不同的计算单元处理子图不好划分的问题。 | ||
搜索关键词: | 一种 应用于 ai 编译器 自动 切分 神经网络 方法 | ||
【主权项】:
暂无信息
下载完整专利技术内容需要扣除积分,VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于开放智能机器(上海)有限公司,未经开放智能机器(上海)有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/patent/201911312625.0/,转载请声明来源钻瓜专利网。