一种面向神经网络模型计算的中间表示方法和装置
授权
摘要
本发明公开了一种面向神经网络模型计算的中间表示方法和装置,包括如下步骤:S1:解析输入的模型文件以获取神经网络的拓扑结构信息;S2:构建逻辑计算图;S21:推导逻辑计算图中每个算子的物理布局信息;S22:推导逻辑计算图中每个算子的元属性;S23:推导逻辑计算图中每个算子的输入输出逻辑张量的描述信息;S3:构建物理计算图;S31:生成物理计算图;等步骤,本发明公开的基于元属性的用于神经网络模型计算的中间表示从算子层面原生地支持数据并行和模型并行以及流水并行。本发明公开的面向神经网络模型计算的中间表示方法和装置以计算表达式为基本单元,以张量作为整个计算表达式组成的计算图中流动的数据,以构图的方式实现神经网络模型的计算过程。
基本信息
专利标题 :
一种面向神经网络模型计算的中间表示方法和装置
专利标题(英):
暂无
公开(公告)号 :
CN114186687A
申请号 :
CN202210144108.2
公开(公告)日 :
2022-03-15
申请日 :
2022-02-17
授权号 :
CN114186687B
授权日 :
2022-05-17
发明人 :
王宏升华炜郏维强鲍虎军
申请人 :
之江实验室
申请人地址 :
浙江省杭州市余杭区之江实验室南湖总部
代理机构 :
北京志霖恒远知识产权代理事务所(普通合伙)
代理人 :
奚丽萍
优先权 :
CN202210144108.2
主分类号 :
G06N3/08
IPC分类号 :
G06N3/08 G06N3/063 G06F9/50
IPC结构图谱
G
G部——物理
G06
计算;推算或计数
G06N
基于特定计算模型的计算机系统
G06N3/00
基于生物学模型的计算机系统
G06N3/02
采用神经网络模型
G06N3/08
学习方法
法律状态
2022-05-17 :
授权
2022-04-01 :
实质审查的生效
IPC(主分类) : G06N 3/08
申请日 : 20220217
申请日 : 20220217
2022-03-15 :
公开
注:本法律状态信息仅供参考,即时准确的法律状态信息须到国家知识产权局办理专利登记簿副本。
文件下载
暂无PDF文件可下载