发明公开
- 专利标题: 张量切分模式的确定方法、装置、计算机设备及介质
-
申请号: CN202210415394.1申请日: 2022-04-20
-
公开(公告)号: CN114723014A公开(公告)日: 2022-07-08
- 发明人: 万学磊 , 田野
- 申请人: 上海燧原科技有限公司
- 申请人地址: 上海市浦东新区中国(上海)自由贸易试验区临港新片区业盛路188号A-522室
- 专利权人: 上海燧原科技有限公司
- 当前专利权人: 上海燧原科技有限公司
- 当前专利权人地址: 上海市浦东新区中国(上海)自由贸易试验区临港新片区业盛路188号A-522室
- 代理机构: 北京品源专利代理有限公司
- 代理商 王瑞云
- 主分类号: G06N3/04
- IPC分类号: G06N3/04 ; G06N3/08 ; G06N3/063
摘要:
本发明实施例公开了一种张量切分模式的确定方法、装置、计算机设备及介质。该方法包括获取包括多个计算算子的待张量切分的计算图;根据与计算算子对应的可选张量切分模式集,确定与计算图对应的包括多个目标计算算子和目标计算算子的目标张量切分模式的数据条目集;根据各目标张量切分模式,计算各数据条目分别对应的计算耗时;根据在数据条目集中选取的多个数据条目对,生成多个交叉数据条目,并根据各所述交叉数据条目的计算耗时,更新数据条目集;在更新后的数据条目集中获取目标数据条目,从而确定计算图中各计算算子的张量切分模式。本发明实施例的技术方案提供一种可适用复杂场景,能够高效确定算子较优切分模式的方法。