[實用新型]神經網絡計算模組

有效

• 申請號: 201721254418.0
• 申請日: 2017-09-27
• 公開（公告）號: CN207895435U
• 公開（公告）日: 2018-09-21
• 發明（設計）人: 奧利維爾·泰馬姆;拉維·納拉亞納斯瓦米;哈西特·哈伊坦;禹同爀
• 申請（專利權）人: 谷歌有限責任公司
• 主分類號: G06F13/28
• 分類號: G06F13/28;G06N3/063;G06F9/302;G06F9/30;G06N3/04
• 代理公司: 中原信達知識產權代理有限責任公司 11219
• 代理人: 周亞榮;安翔

1.一種用于加速張量計算的計算單元，所述計算單元包括：

具有第一數據寬度的第一存儲器群，所述第一存儲器群用于存儲輸入激活和輸出激活中的至少一個；

具有比所述第一數據寬度大的第二數據寬度的第二存儲器群，所述第二存儲器群用于存儲在執行計算時使用的一個或者多個參數；

至少一個cell，所述至少一個cell包括從所述第二存儲器群接收參數并且執行計算的至少一個乘法累加(“MAC”)運算器；

與至少所述第一存儲器群進行數據通信的第一遍歷單元，所述第一遍歷單元被配置為向所述第一存儲器群提供控制信號以使輸入激活被提供至能夠由所述乘法累加運算器訪問的數據總線；以及

其中，所述計算單元通過所述乘法累加運算器執行與數據陣列中的至少一個元素相關聯的一個或者多個計算，所述一個或多個計算至少包括從所述數據總線接收的所述輸入激活與從所述第二存儲器群接收的參數的乘法運算。

2.根據權利要求1所述的計算單元，進一步包括第二遍歷單元，所述第二遍歷單元被配置為基于從所述計算單元外部的源接收的指令來訪問所述第一存儲器群的至少一個存儲器位置和所述第二存儲器群的至少一個存儲器位置。

3.根據權利要求1所述的計算單元，其中，由所述計算單元執行的所述一個或者多個計算包括使用移位寄存器將所述一個或多個輸出激活移位到所述第一存儲器群。

4.根據權利要求1所述的計算單元，進一步包括延伸到所述計算單元外部的環形總線的部分，其中，所述環形總線在所述第一存儲器群與另一相鄰計算單元的存儲器群之間以及所述第二存儲器群和所述另一相鄰計算單元的所述存儲器群之間提供數據路徑。

5.根據權利要求1所述的計算單元，其中，所述第二存儲器群被配置為存儲部分和以及一個或者多個池化層輸入中的至少一種。