一种卷积运算电路及其运算方法

基本信息

申请号 CN202111193721.5 申请日 -
公开(公告)号 CN113869498A 公开(公告)日 2021-12-31
申请公布号 CN113869498A 申请公布日 2021-12-31
分类号 G06N3/04(2006.01)I;G06F17/15(2006.01)I 分类 计算;推算;计数;
发明人 林广栋;陆俊峰;黄光红;顾大晔 申请(专利权)人 安徽芯纪元科技有限公司
代理机构 合肥律众知识产权代理有限公司 代理人 赵娟
地址 230000安徽省合肥市高新区习友路3366号博微产业园系统协同中心7楼
法律状态 -

摘要

摘要 本发明提出一种卷积运算电路及其运算方法,卷积运算电路由PE阵列组成,同时支持矩阵乘法运算和二维卷积运算;当执行矩阵乘法运算时,PE分别接收来自行方向上的左矩阵行输入和来自列方向上的右矩阵列输入,将两者相乘后累加到本地累加寄存器;当执行二维卷积运算时,PE接收来自行方向上的输入特征图输入,将其与内部存储的卷积核权重相乘后与上一行PE输出的部分和进行累加,再向下一行输出累加后的部分和;本发明在PE内部设置多个权重寄存器,把多通道的卷积核权重折叠式地存储在一列PE内部,在PE阵列的底部设置先进先出存储器,用于存储中间计算结果,避免反复在存储器和PE之间传输权重和中间计算结果,提高运算效率。