摘要
本发明公开了一种基于特征图稀疏性的卷积神经网络加速器。所述卷积神经网络加速器包括输入特征图编码模块、权重编码模块、数据流控制模块、稀疏矩阵计算模块和总线;总线分别连接数据流控制模块、输入特征图编码模块、权重编码模块和稀疏矩阵计算模块;输入特征图编码模块根据特征图中的0元素,对特征图编码;权重编码模块根据输入特征图编码模块的编码信息,为稀疏矩阵计算模块提供对应的权重数据;数据流控制模块根据寄存器信息控制其余模块的工作模式;稀疏矩阵计算模块使用输入特征图编码模块和权重编码模块提供的数据进行卷积计算。本发明可以将加速器切换为利用权重中的稀疏性,可以灵活应用,对稀疏权重加以支持。
- 单位