张量计算代码优化方法、装置、设备及介质
首页 企业 产品 技术 资讯 图库 视频 需求 会议 活动 产业
张量计算代码优化方法、装置、设备及介质
来源:北方有色网
访问:1035
简介: 本申请提供一种张量计算代码优化方法、装置、电子设备及介质。其中,所述方法包括:分析张量计算代码的循环特征和计算图特征,得到相应的循环信息和计算图信息,根据所述循环信息和计算图信息以及预设优化方法,生成优化空间,该优化空间中每个空间点表示一个预设优化方法组合及参数选择。基于模拟退火算法和强化学习算法,在优化空间中搜索并确定目标空间点,根据所述目标空间点对应的预设优化方法组合及参数选择,对所述张量计算代码实施优化,从而可以快速完成张量计算代码的自动优化,提高张量计算代码运行效率,对于编程开发人员来说,可以免除开发算子的人力投入,并且取得相对较好的性能,可以减少成本并提高开发效率。
本申请提供一种张量计算代码优化方法、装置、电子设备及介质。其中,所述方法包括:分析张量计算代码的循环特征和计算图特征,得到相应的循环信息和计算图信息,根据所述循环信息和计算图信息以及预设优化方法,生成优化空间,该优化空间中每个空间点表示一个预设优化方法组合及参数选择。基于模拟退火算法和强化学习算法,在优化空间中搜索并确定目标空间点,根据所述目标空间点对应的预设优化方法组合及参数选择,对所述张量计算代码实施优化,从而可以快速完成张量计算代码的自动优化,提高张量计算代码运行效率,对于编程开发人员来说,可以免除开发算子的人力投入,并且取得相对较好的性能,可以减少成本并提高开发效率。
0
0
0
0
0
         
标签:化学分析
广州铭谦选矿设备有限公司宣传
广州铭谦选矿设备有限公司宣传
相关技术
评论(0条)
200/200
牛津仪器科技(上海)有限公司宣传
发布
技术

顶部
北方有色网-互联网服务平台-关于我们
Copyright 2025 China-mcc.com All Rights Reserved
备案号:京ICP备11044340号-3
电信业务经营许可证编号:京B2-20242293
京公网安备 11010702002294号