Invention Publication
- Patent Title: 张量切分模式的确定方法、装置、计算机设备及介质
-
Application No.: CN202210415394.1Application Date: 2022-04-20
-
Publication No.: CN114723014APublication Date: 2022-07-08
- Inventor: 万学磊 , 田野
- Applicant: 上海燧原科技有限公司
- Applicant Address: 上海市浦东新区中国(上海)自由贸易试验区临港新片区业盛路188号A-522室
- Assignee: 上海燧原科技有限公司
- Current Assignee: 上海燧原科技股份有限公司
- Current Assignee Address: 201306 上海市浦东新区中国(上海)自由贸易试验区临港新片区业盛路188号A-522室
- Agency: 北京品源专利代理有限公司
- Agent 王瑞云
- Main IPC: G06N3/04
- IPC: G06N3/04 ; G06N3/08 ; G06N3/063

Abstract:
本发明实施例公开了一种张量切分模式的确定方法、装置、计算机设备及介质。该方法包括获取包括多个计算算子的待张量切分的计算图;根据与计算算子对应的可选张量切分模式集,确定与计算图对应的包括多个目标计算算子和目标计算算子的目标张量切分模式的数据条目集;根据各目标张量切分模式,计算各数据条目分别对应的计算耗时;根据在数据条目集中选取的多个数据条目对,生成多个交叉数据条目,并根据各所述交叉数据条目的计算耗时,更新数据条目集;在更新后的数据条目集中获取目标数据条目,从而确定计算图中各计算算子的张量切分模式。本发明实施例的技术方案提供一种可适用复杂场景,能够高效确定算子较优切分模式的方法。
Information query