神經(jīng)網(wǎng)絡(luò)模型的編譯方法和裝置、電子設(shè)備和存儲介質(zhì)

基本信息

申請?zhí)?/td> CN202111595178.1 申請日 -
公開（公告）號 CN114492730A 公開（公告）日 2022-05-13
申請公布號 CN114492730A 申請公布日 2022-05-13
分類號 G06N3/04（2006.01）I;G06N3/08（2006.01）I 分類 計算；推算；計數(shù);
發(fā)明人 李軍 申請（專利權(quán)）人 北京地平線信息技術(shù)有限公司
代理機構(gòu) 北京思源智匯知識產(chǎn)權(quán)代理有限公司 代理人 -
地址 100094北京市海淀區(qū)豐豪東路9號院2號樓3層1單元301
法律狀態(tài) -

摘要

摘要 本公開實施例公開了一種神經(jīng)網(wǎng)絡(luò)模型的編譯方法和裝置、電子設(shè)備和存儲介質(zhì)，其中，方法包括：針對待編譯神經(jīng)網(wǎng)絡(luò)模型對應(yīng)的第一模型中的至少一個逐點卷積層中的第一逐點卷積層，基于預(yù)設(shè)張量折疊規(guī)則對第一逐點卷積層的原輸入張量進行數(shù)據(jù)結(jié)構(gòu)變換，獲得變換后的第一輸入張量，使得第一輸入張量在至少一個維度上的尺寸與乘法器的對齊要求的尺寸的第一距離小于原輸入張量對應(yīng)的第二距離；基于第一輸入張量獲得第一逐點卷積層對應(yīng)的優(yōu)化后的第二逐點卷積層；基于第二逐點卷積層獲得第一模型對應(yīng)的優(yōu)化模型；根據(jù)優(yōu)化模型生成待編譯神經(jīng)網(wǎng)絡(luò)模型對應(yīng)的二進制的目標(biāo)模型。本公開實施例可以有效減少對齊填充量，從而提高乘法器的有效利用率。