張量切分模式的確定方法、裝置、計算機設(shè)備及介質(zhì)

基本信息

申請?zhí)?/td> CN202210415394.1 申請日 -
公開（公告）號 CN114723014A 公開（公告）日 2022-07-08
申請公布號 CN114723014A 申請公布日 2022-07-08
分類號 G06N3/04（2006.01）I;G06N3/08（2006.01）I;G06N3/063（2006.01）I 分類 計算；推算；計數(shù);
發(fā)明人 萬學(xué)磊;田野 申請（專利權(quán)）人 上海燧原科技有限公司
代理機構(gòu) 北京品源專利代理有限公司 代理人 -
地址 201306上海市浦東新區(qū)中國（上海）自由貿(mào)易試驗區(qū)臨港新片區(qū)業(yè)盛路188號A-522室
法律狀態(tài) -

摘要

摘要 本發(fā)明實施例公開了一種張量切分模式的確定方法、裝置、計算機設(shè)備及介質(zhì)。該方法包括獲取包括多個計算算子的待張量切分的計算圖；根據(jù)與計算算子對應(yīng)的可選張量切分模式集，確定與計算圖對應(yīng)的包括多個目標計算算子和目標計算算子的目標張量切分模式的數(shù)據(jù)條目集；根據(jù)各目標張量切分模式，計算各數(shù)據(jù)條目分別對應(yīng)的計算耗時；根據(jù)在數(shù)據(jù)條目集中選取的多個數(shù)據(jù)條目對，生成多個交叉數(shù)據(jù)條目，并根據(jù)各所述交叉數(shù)據(jù)條目的計算耗時，更新數(shù)據(jù)條目集；在更新后的數(shù)據(jù)條目集中獲取目標數(shù)據(jù)條目，從而確定計算圖中各計算算子的張量切分模式。本發(fā)明實施例的技術(shù)方案提供一種可適用復(fù)雜場景，能夠高效確定算子較優(yōu)切分模式的方法。