[發明專利]圖重寫的處理方法及裝置、計算設備及可讀介質

在審

• 申請號: 201811550198.5
• 申請日: 2018-12-18
• 公開（公告）號: CN111340175A
• 公開（公告）日: 2020-06-26
• 發明（設計）人: 李天平;孫曉明
• 申請（專利權）人: 賽靈思公司
• 主分類號: G06N3/04
• 分類號: G06N3/04
• 代理公司: 北京展翼知識產權代理事務所(特殊普通合伙) 11452
• 代理人: 張陽

1.一種圖重寫的處理方法，其特征在于，所述方法包括；

對網絡部署框架的前端的神經網絡模型進行解析，獲取所述神經網絡模型對應的原始圖，所述原始圖包括數個單運算的操作符算子；

利用預先生成的圖重寫接口對所述原始圖進行重寫處理，得到所述網絡部署框架的后端支持處理的重寫圖，所述優化操作符算子為所述原始圖中相鄰的至少兩個所述單運算的操作符算子的優化組合。

2.根據權利要求1所述的方法，其特征在于，利用預先生成的圖重寫接口對所述原始圖進行重寫處理，得到所述網絡部署框架的后端支持處理的重寫圖之前，所述方法還包括：

根據所述網絡部署框架和所述后端支持的計算/編譯能力生成所述圖重寫接口。

3.根據權利要求2所述的方法，其特征在于，根據所述網絡部署框架和所述后端支持的計算/編譯能力生成所述圖重寫接口，包括：

根據所述后端支持的計算/編譯能力和所述網絡部署框架所提供的操作符類型，在模版文件中定義用于圖重寫的多種模式；

對所述模版文件中的所述多種模式進行解析，得到模式列表；

生成所述模式列表中的各所述模式匹配的代碼。

4.根據權利要求3所述的方法，其特征在于，根據所述后端支持的計算/編譯能力和所述網絡部署框架所提供的操作符類型，在模版文件中定義用于圖重寫的多種優化模式之后，對所述模版文件中的所述多種模式進行解析，得到模式列表之前，所述方法還包括：

對各所述模式進行合法性驗證，確定各所述模式合法。

5.根據權利要求3所述的方法，其特征在于，根據所述網絡部署框架和所述后端支持的計算/編譯能力生成所述圖重寫接口，還包括：

配置模式定義和/或模式解析的功能支持信息。

6.根據權利要求3所述的方法，其特征在于，根據所述后端支持的計算/編譯能力和所述網絡部署框架所提供的操作符類型，在模版文件中定義用于圖重寫的多種模式，包括：

根據所述網絡部署框架所提供的操作符類型，在所述模版文件中定義數種單運算對應的簡單模式；

根據所述后端支持的計算/編譯能力和所述網絡部署框架所提供的操作符類型，在所述模版文件中定義數種優化模式，各所述優化模式對應至少兩種單運算的組合。

7.根據權利要求3-6任一所述的方法，其特征在于，根據所述后端支持的計算/編譯能力和所述網絡部署框架所提供的操作符類型，在模版文件中定義用于圖重寫的多種模式，包括：

確定在所述模本文件中采用預設的模版文件格式定義用于圖重寫的所述多種模式的模式定義格式；

根據所述后端支持的計算/編譯能力和所述網絡部署框架所提供的操作符類型，采用所述模式定義格式，在所述模板文件中定義所述多種模式。

8.根據權利要求3-6任一所述的方法，其特征在于，利用預先生成的圖重寫接口對所述原始圖進行重寫處理，得到所述網絡部署框架的后端支持處理的重寫圖，包括：

根據所述圖重寫接口中生成有匹配的代碼的各種優化模式，判斷所述原始圖中相鄰的至少兩個所述單運算的操作符算子是否能夠進行優化組合；

若能，將所述原始圖中相鄰的至少兩個所述單運算的操作符算子優化組合為所述優化操作符算子，得到所述重寫圖。

9.一種圖重寫的處理裝置，其特征在于，所述裝置包括；

原始圖獲取模塊，用于對網絡部署框架的前端的神經網絡模型進行解析，獲取所述神經網絡模型對應的原始圖，所述原始圖包括數個單運算的操作符算子；

重寫處理模塊，用于利用預先生成的圖重寫接口對所述原始圖進行重寫處理，得到所述網絡部署框架的后端支持處理的重寫圖，所述優化操作符算子為所述原始圖中相鄰的至少兩個所述單運算的操作符算子的優化組合。

10.根據權利要求9所述的裝置，其特征在于，所述裝置還包括：

生成模塊，用于根據所述網絡部署框架和所述后端支持的計算/編譯能力生成所述圖重寫接口。