[发明专利]神经网络加速方法、装置、神经网络加速芯片及存储介质

权利要求书

1.一种神经网络加速方法，其特征在于，应用于神经网络加速芯片，该方法包括：

针对待加速的神经网络，进行下述步骤，直至确定所述神经网络加速完成：

采用待加速的当前层的参数对所述当前层进行加速处理，并调度所述当前层的下一层的参数；

当所述当前层加速处理完成时，将所述下一层确定为待加速的当前层进行加速处理。

2.如权利要求1所述的方法，其特征在于，如果所述待加速的当前层为最后一层，所述调度所述当前层的下一层的参数包括：

调度第一层的参数。

3.如权利要求1所述的方法，其特征在于，所述调度所述当前层的下一层的参数包括：

调度片上存储器中保存的所述当前层的下一层的参数。

4.如权利要求3所述的方法，其特征在于，所述调度片上存储器中保存的所述当前层的下一层的参数包括：

通过REG文件，调度片上存储器中保存的所述当前层的下一层的参数。

5.如权利要求3或4所述的方法，其特征在于，所述采用待加速的当前层的参数对所述当前层进行加速处理，并调度所述当前层的下一层的参数之前，所述方法还包括：

从待加速的神经网络中提取加速处理所需的每层的参数，并保存到所述片上存储器中。

6.如权利要求3或4所述的方法，其特征在于，所述片上存储器包括只读存储器ROM。

7.一种神经网络加速装置，其特征在于，应用于神经网络加速芯片，该装置包括：

加速调度模块，用于针对待加速的神经网络，进行下述步骤，直至确定所述神经网络加速完成：采用待加速的当前层的参数对所述当前层进行加速处理，并调度所述当前层的下一层的参数；

确定模块，用于当所述当前层加速处理完成时，将所述下一层确定为待加速的当前层进行加速处理。

8.如权利要求7所述的装置，其特征在于，所述加速调度模块，具体用于如果所述待加速的当前层为最后一层，调度第一层的参数。

9.如权利要求7所述的装置，其特征在于，所述加速调度模块，具体用于调度片上存储器中保存的所述当前层的下一层的参数。

10.如权利要求9所述的装置，其特征在于，所述加速调度模块，具体用于通过REG文件，调度片上存储器中保存的所述当前层的下一层的参数。

11.如权利要求9或10所述的装置，其特征在于，所述装置还包括：

提取保存模块，用于从待加速的神经网络中提取加速处理所需的每层的参数，并保存到所述片上存储器中。

12.如权利要求9或10所述的装置，其特征在于，所述片上存储器包括只读存储器ROM。

13.一种神经网络加速芯片，其特征在于，包括：处理器、通信接口、存储器和通信总线，其中，处理器，通信接口，存储器通过通信总线完成相互间的通信；

所述存储器中存储有计算机程序，当所述程序被所述处理器执行时，使得所述处理器执行权利要求1～6任一项所述方法的步骤。

14.一种计算机可读存储介质，其特征在于，其存储有可由神经网络加速芯片执行的计算机程序，当所述程序在所述神经网络加速芯片上运行时，使得所述神经网络加速芯片执行权利要求1～6任一项所述方法的步骤。