本申請提供一種張量計算代碼優化方法、裝置、電子設備及介質。其中,所述方法包括:分析張量計算代碼的循環特征和計算圖特征,得到相應的循環信息和計算圖信息,根據所述循環信息和計算圖信息以及預設優化方法,生成優化空間,該優化空間中每個空間點表示一個預設優化方法組合及參數選擇?;谀M退火算法和強化學習算法,在優化空間中搜索并確定目標空間點,根據所述目標空間點對應的預設優化方法組合及參數選擇,對所述張量計算代碼實施優化,從而可以快速完成張量計算代碼的自動優化,提高張量計算代碼運行效率,對于編程開發人員來說,可以免除開發算子的人力投入,并且取得相對較好的性能,可以減少成本并提高開發效率。
聲明:
“張量計算代碼優化方法、裝置、設備及介質” 該技術專利(論文)所有權利歸屬于技術(論文)所有人。僅供學習研究,如用于商業用途,請聯系該技術所有人。
我是此專利(論文)的發明人(作者)