用于LSTM网络的硬件加速器

基本信息

申请号 CN201810157774.3 申请日 -
公开(公告)号 CN110197262B 公开(公告)日 2021-07-30
申请公布号 CN110197262B 申请公布日 2021-07-30
分类号 G06N3/063;G06N3/04 分类 计算;推算;计数;
发明人 康君龙;李於彬 申请(专利权)人 北京深鉴智能科技有限公司
代理机构 北京展翼知识产权代理事务所(特殊普通合伙) 代理人 屠长存
地址 100029 北京市朝阳区安定路5号院5号楼6层101号10层01-19单元
法律状态 -

摘要

摘要 本发明公开了一种用于长短期记忆(LSTM)网络的硬件加速器。LSTM计算模块对T个有序输入进行前向LSTM计算,得到分别对应于所述T个有序输入的T个第一输出,并对所述T个有序输入进行后向LSTM计算,得到分别对应于所述T个有序输入的T个第二输出,其中,T为正整数。第一缓存用于缓存所述第一输出和所述第二输出。组合模块对所述第一缓存中缓存的对应于同一个输入的第一输出和第二输出执行组合运算,以得到对应于所述同一个输入的组合输出。由此,可以提高双向LSTM计算性能,降低响应延时,实现对双向LSTM网络计算的加速效果。