买专利卖专利找龙图腾,真高效! 查专利查商标用IPTOP,全免费!专利年费监控用IP管家,真方便!
申请/专利权人:广州希姆半导体科技有限公司
摘要:本发明实施例公开了一种计算图算子切分的方法及装置。所述计算图算子切分的方法会确定与计算图中各算子的输出张量对应的最小切分个数,并根据最小切分个数确定与各算子对应的目标切分参数集合,再根据计算图将各目标切分参数集合中的切分参数进行组合以确定多个候选切分方案,进而在多个所述候选切分方案中确定目标切分方案。其中,所述最小切分个数用于表征在切分所述输出张量时的最少切分个数,所述切分参数用于表征在对应维度切分所述算子的输出张量时的切分个数,所述目标切分方案用于指示切分计算图中各算子的输出张量。通过所述方法可以提高所确定切分方案的全局适应性,改善AI编译器的图层优化效果。
主权项:1.一种计算图算子切分的方法,其特征在于,所述方法包括:确定与计算图中各算子的输出张量对应的最小切分个数,所述最小切分个数用于表征在切分所述输出张量时的最少切分个数;根据所述最小切分个数确定与各所述算子对应的目标切分参数集合,所述目标切分参数集合包括至少一个切分参数,各所述切分参数具有对应的维度,所述切分参数用于表征在对应维度切分所述算子的输出张量时的切分个数;根据所述计算图将各所述目标切分参数集合中的切分参数进行组合以确定多个候选切分方案;在多个所述候选切分方案中确定目标切分方案,所述目标切分方案用于指示切分所述计算图中各所述算子的输出张量。
全文数据:
权利要求:
百度查询: 广州希姆半导体科技有限公司 计算图算子切分的方法及装置
免责声明
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。