張量切分模式的確定方法、裝置、計(jì)算機(jī)設(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 分類 計(jì)算；推算；計(jì)數(shù);
發(fā)明人 萬學(xué)磊;田野 申請（專利權(quán)）人 上海燧原科技有限公司
代理機(jī)構(gòu) 北京品源專利代理有限公司 代理人 -
地址 201306上海市浦東新區(qū)中國（上海）自由貿(mào)易試驗(yàn)區(qū)臨港新片區(qū)業(yè)盛路188號A-522室
法律狀態(tài) -

摘要

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