[发明专利]卷积计算加速方法和装置

权利要求书

1.一种卷积计算加速方法，其特征在于，包括：

判断卷积类型，当卷积类型为深度优先卷积时，获取当前深度时序；

将所述当前深度时序对应的卷积核中的列像素和特征图中的行像素输入至卷积计算单元；所述卷积计算单元是由乘加器构成的阵列；

若，则将卷积核的第一行像素平移至最后一行；其中，为所述卷积核的行列数，即当前深度时序能够被卷积核的行列数整除；

若，且，将所述卷积计算单元中第行特征作为第行深度优先卷积的输出特征；

所述将所述当前深度时序对应的所述卷积核中的列像素和特征图中的行像素输入至卷积计算单元，之前还包括：

获取所述卷积核中的第列像素，作为所述当前深度时序对应的所述卷积核中的列像素；

以及，获取所述特征图中第行的第至第个像素作为所述当前深度时序对应的所述特征图的行像素；其中，卷积计算单元的列数为，所述特征图的列数为；

所述将所述当前深度时序对应的所述卷积核中的列像素和特征图中的行像素输入至卷积计算单元，包括：

在垂直方向上，将当前深度时序对应的卷积核中的列像素中的每一像素分别输入卷积计算单元中的一行，在水平方向上，将当前深度时序对应的特征图中的行像素中的每一像素分别输入卷积计算单元中的一列；

还包括：

当所述卷积类型为二维卷积时，获取当前二维时序；

将所述当前二维时序对应的每一所述卷积核中的像素和所述特征图中的行像素输入至所述卷积计算单元；

二维卷积结束后，将所述卷积计算单元中前行特征作为二维卷积输出特征；其中，为所述二维卷积中的卷积核数量；

所述将所述当前二维时序对应的每一所述卷积核中的像素和所述特征图中的行像素输入至所述卷积计算单元，之前还包括：

获取每一所述卷积核中第行的第个像素作为所述当前二维时序对应的每一所述卷积核中的像素；

以及，获取所述特征图中第行的第至第个像素作为当前二维时序对应的所述特征图的行像素；其中，卷积计算单元的列数为，所述特征图的列数为。

2.根据权利要求1所述的卷积计算加速方法，其特征在于，还包括：

当所述卷积类型为深度优先卷积时，关闭所述卷积计算单元中每一行号大于的所述乘加器。

3.一种卷积计算加速装置，其特征在于，包括：

深度时序获取单元，用于判断卷积类型，当卷积类型为深度优先卷积时，获取当前深度时序；

深度卷积计算单元，用于将所述当前深度时序i对应的卷积核中的列像素和特征图中的行像素输入至卷积计算单元；所述卷积计算单元是由乘加器构成的阵列；

卷积核更新单元，用于若，则将卷积核的第一行像素平移至最后一行；其中，为所述卷积核的行列数，即当前深度时序能够被卷积核的行列数整除；

特征输出单元，用于若，且，将所述卷积计算单元中第行特征作为第行深度优先卷积的输出特征；

深度列像素获取单元，用于获取所述卷积核中的第列像素，作为所述当前深度时序对应的所述卷积核中的列像素；

以及，深度行像素获取单元，用于获取所述特征图中第行的第至第个像素作为所述当前深度时序对应的所述特征图的行像素；其中，卷积计算单元的列数为，所述特征图的列数为；

所述深度卷积计算单元具体用于：

在垂直方向上，将当前深度时序对应的卷积核中的列像素中的每一像素分别输入卷积计算单元中的一行，在水平方向上，将当前深度时序对应的特征图中的行像素中的每一像素分别输入卷积计算单元中的一列；

二维卷积计算单元，用于：

当所述卷积类型为二维卷积时，获取当前二维时序；

将所述当前二维时序对应的每一所述卷积核中的像素和所述特征图中的行像素输入至所述卷积计算单元；

二维卷积结束后，将所述卷积计算单元中前行特征作为二维卷积输出特征；其中，为所述二维卷积中的卷积核数量；

所述二维卷积计算单元还用于：

获取每一所述卷积核中第行的第个像素作为所述当前二维时序对应的每一所述卷积核中的像素；

以及，获取所述特征图中第行的第至第个像素作为当前二维时序对应的所述特征图的行像素；其中，卷积计算单元的列数为，所述特征图的列数为。