[发明专利]一种神经网络加速系统

权利要求书

1.一种神经网络加速系统，其特征在于，包括：

信息配置模块，由外部设置信息配置模块的参数信息；

外部存储模块；

和

池化加速单元，分别与信息配置模块和外部存储模块相连，所述池化加速单元包括：

控制模块，所述控制模块根据所述信息配置模块的参数信息得到张量块的信息以及地址信息，所述张量块包括输入张量块和输出张量块；

输入张量缓存模块，与所述控制模块相连，根据由控制模块得到的输入张量块地址信息缓存从外部存储模块的输入张量块；

过程张量缓存模块，与所述控制模块相连，根据由控制模块得到的输出张量块地址信息缓存沿宽度方向池化的输出张量块，并提供沿高度方向池化的输入数据；

池化器阵列模块，分别与输入张量缓存模块和过程张量缓存模块相连，进行宽度方向的池化和高度方向的池化，并沿通道方向进行并行计算。

2.根据权利要求1所述的一种神经网络加速系统，其特征在于，

所述参数信息包括池化层基本信息、输入张量块和输出张量块的相关信息；

所述池化层基本信息包括：池化类型、池化窗口高度、池化窗口宽度、池化高度方向步长、池化宽度方向步长、输入数据高度方向边界补零信息和输入数据宽度方向边界补零信息；

输入张量块相关信息包括：输入块位置序号、输入张量块高度、输入张量块宽度和输入通道数；

输出张量块相关信息包括：输出张量块高度和输出张量块宽度。

3.根据权利要求1所述的一种神经网络加速系统，其特征在于，

所述外部存储模块的存储方式包括：

将沿通道方向拆分成两个维度，分别为低维度的PDI和高维度的T；

数据沿通道方向进行存储；

当通道方向的数据存储达到PDI边界时，沿宽度方向进行存储；

当宽度方向的数据存储达到宽度边界时，沿高度方向进行存储；

当高度方向的数据存储达到高度边界时，沿通道方向向高纬度T进行存储。

4.根据权利要求3所述的一种神经网络加速系统，其特征在于，

所述控制模块根据所述信息配置模块的参数信息得到张量块的信息包括，基于所述参数信息和输出张量块元素的信息得到输入张量块元素的高度索引，所述输出张量块元素的信息包括由输出张量块进行四个维度的遍历得到的输出张量块元素的位置信息，所述输入张量块元素的高度索引表示为：

其中，n_tih为输入张量块元素的高度索引，n_toh为输出张量块元素的高度索引，S_h为池化高度方向步长，k_h为池化窗口高度，P_h为输入数据高度方向边界补零信息，max_tih为输入张量块元素的高度索引的最大值。

5.根据权利要求4所述的一种神经网络加速系统，其特征在于，

所述控制模块根据所述信息配置模块的参数信息得到张量块的信息还包括，基于所述参数信息和输出张量块元素的信息得到输入张量块元素的宽度索引，所述输出张量块元素的信息包括由输出张量块进行四个维度的遍历得到的输出张量块元素的位置信息，所述输入张量块元素的宽度索引表示为：

其中，n_tiw为输入张量块元素的宽度索引，n_tow为输出张量块元素的宽度索引，S_w为池化宽度方向步长，k_w为池化窗口宽度，P_w为输入数据宽度方向边界补零信息，max_tiw输入张量块元素的宽度索引的最大值。