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