厦门壹普智慧科技有限公司赵凯荣获国家专利权
买专利卖专利找龙图腾,真高效! 查专利查商标用IPTOP,全免费!专利年费监控用IP管家,真方便!
龙图腾网获悉厦门壹普智慧科技有限公司申请的专利面向通用张量处理器的算子并行分割方法及系统获国家发明授权专利权,本发明授权专利权由国家知识产权局授予,授权公告号为:CN117077744B 。
龙图腾网通过国家知识产权局官网在2025-09-23发布的发明授权授权公告中获悉:该发明授权的专利申请号/专利号为:202311052377.7,技术领域涉及:G06N3/063;该发明授权面向通用张量处理器的算子并行分割方法及系统是由赵凯荣;罗闳訚;何日辉;周志新;汤梦饶设计研发完成,并于2023-08-21向国家知识产权局提交的专利申请。
本面向通用张量处理器的算子并行分割方法及系统在说明书摘要公布了:本发明提出了一种面向通用张量处理器的算子并行分割方法及系统,包括:计算获得算子输出形状;根据算子输出形状分别获取每个维度形状的因数集合;获取算子在TPU中运行时间最短的分割方案,分割方案为因数集合的元素组合,分割方案的总数为因数集合的大小乘积,运行时间的计算表达式为:运行时间=输入读取时间+权重读取时间+计算时间+输出写入时间,输入读取时间的计算以子算子输入大小作为变量,子算子输入大小通过分析神经网络的依赖关系获得;预设解空间阈值,若分割方案的总数不大于解空间阈值,采用枚举算法找出最优分割方案;若大于,则采用启发式算法找出最优分割方案。本发明能够实现基于TPU的神经网络算法自动并行化方案。
本发明授权面向通用张量处理器的算子并行分割方法及系统在权利要求书中公布了:1.一种面向通用张量处理器的算子并行分割方法,其特征在于,包括以下步骤: S1、根据输入形状、卷积核形状和卷积层参数计算获得算子输出形状,所述算子输出形状包括通道输出数据、高度输出数据和宽度输出数据; S2、根据所述算子输出形状分别获取每个维度形状的因数集合,所述因数集合包括通道因数集合、高度因数集合和宽度因数集合; S3、获取算子在TPU中运行时间最短的分割方案,其中所述分割方案为步骤S2中的所述因数集合的元素组合,所述分割方案的总数为步骤S2中的所述因数集合的大小乘积,所述运行时间的计算表达式为: 运行时间=输入读取时间+权重读取时间+计算时间+输出写入时间,式中,所述输入读取时间的计算以子算子输入大小作为变量,所述子算子输入大小通过分析神经网络的依赖关系获得; S4、预设解空间阈值,响应于确定所述分割方案的总数不大于所述解空间阈值,则采用枚举算法遍历解空间找出最优分割方案;响应于确定所述分割方案的总数大于所述解空间阈值,则采用启发式算法找出最优分割方案。
如需购买、转让、实施、许可或投资类似专利技术,可联系本专利的申请人或专利权人厦门壹普智慧科技有限公司,其通讯地址为:361000 福建省厦门市火炬高新区软件园三期诚毅北大街8号1902室A区;或者联系龙图腾网官方客服,联系龙图腾网可拨打电话0551-65771310或微信搜索“龙图腾网”。
以上内容由龙图腾AI智能生成。
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。

皖公网安备 34010402703815号
请提出您的宝贵建议,有机会获取IP积分或其他奖励