一种针对量化后的LSTM的激活函数计算单元

基本信息

申请号 CN201911076853.2 申请日 -
公开(公告)号 CN111047007B 公开(公告)日 2021-07-30
申请公布号 CN111047007B 申请公布日 2021-07-30
分类号 G06N3/04(2006.01)I 分类 计算;推算;计数;
发明人 郑勇;陈柱佳;舒毅 申请(专利权)人 北京中科胜芯科技有限公司
代理机构 南京行高知识产权代理有限公司 代理人 李晓
地址 100044北京市海淀区高梁桥斜街42号院1号楼6层1-609
法律状态 -

摘要

摘要 本发明涉及一种针对量化后的LSTM的激活函数计算单元,其特征在于,包括:sigmoid函数计算单元和tanh函数计算单元;所述tanh函数计算单元包括第一求补码单元、第二求补码单元、幅度比较器、算术运算单元、第一选择器和第二选择器。本发明提供的针对量化后的LSTM的激活函数计算单元,针对LSTM网络的操作数被量化后,根据其网络精度、噪声容限以及线性函数中的偏置系数对操作数的位宽需求确定激活函数计算单元的操作数位宽。针对8 bits位宽的操作数,确定用于近似激活函数的线性函数表达式。针对线性函数表达式设计对应的计算单元。利用booth算法,使用简单的移位相加快速计算系数固定的乘法。