神經(jīng)網(wǎng)絡(luò)編譯器配置方法、裝置、計算機設(shè)備和存儲介質(zhì)

基本信息

申請?zhí)?/td> CN202111267825.6 申請日 -
公開（公告）號 CN113703741B 公開（公告）日 2022-02-22
申請公布號 CN113703741B 申請公布日 2022-02-22
分類號 G06F8/30（2018.01）I;G06N3/063（2006.01）I;G06N3/08（2006.01）I 分類 計算；推算；計數(shù);
發(fā)明人 白楊;呂江波;沈小勇 申請（專利權(quán)）人 上海思謀科技有限公司
代理機構(gòu) 廣州華進(jìn)聯(lián)合專利商標(biāo)代理有限公司 代理人 吳平
地址 518051廣東省深圳市南山區(qū)高新南九道45號三航科技大廈22樓
法律狀態(tài) -

摘要

摘要 本申請涉及一種神經(jīng)網(wǎng)絡(luò)編譯器配置方法、裝置、計算機設(shè)備和存儲介質(zhì)。所述方法包括：獲取目標(biāo)神經(jīng)網(wǎng)絡(luò)模型，以及目標(biāo)神經(jīng)網(wǎng)絡(luò)模型對應(yīng)的初始計算圖；初始計算圖包括多個算子；對多個算子進(jìn)行劃分，得到多個算子集合；確定各算子集合分別對應(yīng)的多種算子類型組合，并獲取各算子類型組合對應(yīng)的算子運行時間，得到各算子集合分別對應(yīng)的多個算子運行時間；將多種算子類型組合中，算子運行時間最小的算子類型組合，作為各算子集合對應(yīng)的目標(biāo)算子類型組合；根據(jù)各算子集合對應(yīng)的目標(biāo)算子類型組合生成目標(biāo)神經(jīng)網(wǎng)絡(luò)模型的目標(biāo)計算圖，根據(jù)目標(biāo)計算圖生成針對于目標(biāo)神經(jīng)網(wǎng)絡(luò)模型的編譯器配置信息。采用本方法能夠提高神經(jīng)網(wǎng)絡(luò)模型編譯器配置優(yōu)化效果。