[发明专利]张量切分模式的确定方法、装置、计算机设备及介质在审
| 申请号： | 202210415394.1 | 申请日： | 2022-04-20 |
| 公开（公告）号： | CN114723014A | 公开（公告）日： | 2022-07-08 |
| 发明（设计）人： | 万学磊;田野 | 申请（专利权）人： | 上海燧原科技有限公司 |
| 主分类号： | G06N3/04 | 分类号： | G06N3/04;G06N3/08;G06N3/063 |
| 代理公司： | 北京品源专利代理有限公司 11332 | 代理人： | 王瑞云 |
| 地址： | 201306 上海市浦东新区中国（上海*** | 国省代码： | 上海;31 |
| 权利要求书： | 查看更多 | 说明书： | 查看更多 |
| 摘要： | 本发明实施例公开了一种张量切分模式的确定方法、装置、计算机设备及介质。该方法包括获取包括多个计算算子的待张量切分的计算图；根据与计算算子对应的可选张量切分模式集，确定与计算图对应的包括多个目标计算算子和目标计算算子的目标张量切分模式的数据条目集；根据各目标张量切分模式，计算各数据条目分别对应的计算耗时；根据在数据条目集中选取的多个数据条目对，生成多个交叉数据条目，并根据各所述交叉数据条目的计算耗时，更新数据条目集；在更新后的数据条目集中获取目标数据条目，从而确定计算图中各计算算子的张量切分模式。本发明实施例的技术方案提供一种可适用复杂场景，能够高效确定算子较优切分模式的方法。 | ||
| 搜索关键词： | 张量 切分 模式 确定 方法 装置 计算机 设备 介质 | ||
【主权项】：
暂无信息
下载完整专利技术内容需要扣除积分，VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息，商用须获得专利权人授权。该专利全部权利属于上海燧原科技有限公司，未经上海燧原科技有限公司许可，擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作，请联系【客服】
本文链接：http://www.vipzhuanli.com/patent/202210415394.1/，转载请声明来源钻瓜专利网。
