[發明專利]用于卷積計算的方法、計算設備和計算機可讀存儲介質

有效

• 申請號: 202011484326.8
• 申請日: 2020-12-16
• 公開（公告）號: CN112614040B
• 公開（公告）日: 2021-09-21
• 發明（設計）人: 不公告發明人
• 申請（專利權）人: 上海壁仞智能科技有限公司
• 主分類號: G06T1/20
• 分類號: G06T1/20;G06F17/15;G06N3/063
• 代理公司: 北京市金杜律師事務所 11256
• 代理人: 黃倩

1.一種用于卷積計算的方法，包括：

如果確定當前周期為初始周期，從輸入緩存向脈動陣列加載n*n像素的數據，所述脈動陣列包括第一預定數目行乘以第二預定數目列的處理單元，所述初始周期與初始權重相關聯，n為自然數；

如果確定當前周期為初始周期之后的其他周期，基于當前周期的順序，向脈動陣列加載一列像素的數據、一行像素的數據或者一個像素的數據；以及

如果確定完成當前周期所對應的數據加載，經由輸出復用單元，在脈動陣列中的一組或多組n*n像素的數據中選擇一組n*n像素的數據，以便提供至通用矩陣乘單元。

2.根據權利要求1所述的方法，其中由輸入緩存向脈動陣列加載n*n像素的數據包括：

經由輸入復用單元，從輸入緩存向脈動陣列加載n*n像素的數據，所述輸入復用單元包括多個第一輸入復用單元和一個第二輸入復用單元。

3.根據權利要求1所述的方法，其中基于當前周期的順序向脈動陣列加載一列像素的數據、一行像素的數據或者一個像素的數據包括：

如果確定當前周期為第一周期，向所述脈動陣列加載第n+1列像素的數據；以及

如果確定當前周期為第二周期，向所述脈動陣列加載第n+2列像素的數據。

4.根據權利要求3所述的方法，其中基于當前周期的順序向脈動陣列加載一列像素的數據、一行像素的數據或者一個像素的數據包括：

如果確定當前周期為第三周期或者第六周期，所述脈動陣列自下而上地脈動所述處理單元的數據；以及

經由輸入復用單元，從輸入緩存向所述脈動陣列加載第n行像素的數據。

5.根據權利要求4所述的方法，其中基于當前周期的順序向脈動陣列加載一列像素的數據、一行像素的數據或者一個像素的數據包括：

如果確定當前周期為第四周期、第五周期、第七周期和第八周期中的一個周期，向脈動陣列加載一個像素的數據。

6.根據權利要求1所述的方法，其中所述第一預定數目為8，所述第二預定數目為8或者11。

7.根據權利要求1所述的方法，其中所加載的n*n像素的數據為8*8像素的數據。

8.一種計算設備，包括：

輸入緩存，用于緩存待加載的數據；

輸入復用單元，用于在初始周期，將來自所述輸入緩存的n*n像素的待加載的數據向脈動陣列加載，以及在所述初始周期之后的其他周期，向脈動陣列加載一列像素的數據、一行像素的數據或者一個像素的數據；

脈動陣列，被配置在所述輸入緩存和通用矩陣乘單元之間，所述脈動陣列包括第一預定數目行乘以第二預定數目列的處理單元；以及

輸出復用單元，用于在每一個周期從所述脈動陣列中的一組或者多組n*n像素的數據中選擇一組n*n像素的數據，以便提供至所述通用矩陣乘單元。

9.根據權利要求8所述的設備 ，其中所述輸入復用單元包括多個第一輸入復用單元和第二輸入復用單元，所述第一預定數目為8，所述第二預定數目為8或者11。

10.一種存儲有計算機指令的計算機可讀存儲介質，其中所述計算機指令用于使所述計算機執行根據權利要求1-7中任一項所述的方法。