← 返回列表

用于执行LSTM神经网络运算的装置和运算方法

申请号: CN202110708810.2
申请人: 上海寒武纪信息科技有限公司
申请日期: 2016年12月30日

摘要文本

一种用于执行LSTM神经网络运算的装置和运算方法。该装置包括直接内存访问单元、指令缓存单元、控制器单元、并列设置的多个数据缓存单元和并列设置的多个数据处理模块,其中所述多个数据处理模块与所述数据缓存单元一一对应,用于从相应数据缓存单元中获取输入数据和运算时所需的权值和偏置,进行LSTM神经网络运算;所述多个数据处理模块之间执行并行运算。本发明采用专用指令运行,运算所需指令数大幅减少,译码开销降低;将权值和偏置缓存,使得数据传输开销降低;本发明不限制具体应用领域,可在诸如语音识别、文本翻译、音乐合成等领域使用,可扩展性强;多个数据处理模块并行运行,显著提高LSTM网络的运算速度。 (来 自 马 克 数 据 网)

专利详细信息

项目 内容
专利名称 用于执行LSTM神经网络运算的装置和运算方法
专利类型 发明授权
申请号 CN202110708810.2
申请日 2016年12月30日
公告号 CN113537480B
公开日 2024年4月2日
IPC主分类号 G06N3/063
权利人 上海寒武纪信息科技有限公司
发明人 陈云霁; 陈小兵; 刘少礼; 陈天石
地址 上海市浦东新区同汇路168号B座6层

专利主权项内容

1.一种用于执行LSTM神经网络运算的装置,其特征在于,包括:并行设置的多个数据缓存单元,每个所述数据缓存单元包括权值、偏置和一份完整的输入数据,其中,每个数据缓存单元中读入的权值和偏置为总的权值和偏置的一部分;其中,所述总的权值和偏置是指从所述装置的外部指定地址读入的权值和偏置;并行设置的多个数据处理模块,与所述多个并行设置的数据缓存单元一一对应,且所述多个数据处理模块之间执行并行运算,每个所述数据处理模块用于从对应数据缓存单元中获取输入数据和运算时所需的权值和偏置,进行LSTM神经网络运算,以得到每个所述数据处理模块中的输出值,并写回到所述数据缓存单元;待所有所述数据处理模块中的输出值写回到数据缓存单元后,将各个数据处理模块中的输出值通过拼接得到最终的输出值。 百度搜索马 克 数 据 网