买专利卖专利找龙图腾,真高效! 查专利查商标用IPTOP,全免费!专利年费监控用IP管家,真方便!
申请/专利权人:深圳云天励飞技术股份有限公司;江苏云天励飞技术有限公司
摘要:本申请实施例提供一种对接TVM的方法及相关设备,该方法包括:采用TVM根据目标模型生成第一计算图,其中,所述目标模型用于芯片开发;根据所述第一计算图生成第二计算图,其中,所述第二计算图的结构为所述芯片开发使用的计算图结构,所述第二计算图为芯片开发环境的输入。采用本申请实施例,能够减少TVM引入对芯片开发环境的运算资源需求,提升运行速率,减少芯片开发环境的运行时间。
主权项:1.一种对接矢量虚拟机TVM的装置,其特征在于,应用于电子设备,所述装置包括:TVM修正模块,用于采用TVM根据目标模型生成第一计算图,其中,所述目标模型用于芯片开发;计算图产生模块,所述计算图产生模块包括TVM算子参数模板列表和计算图解析单元,所述TVM算子参数模板列表根据所述TVM使用的算子得到,所述计算图解析单元,用于:根据所述TVM算子参数模板列表对所述第一计算图进行解析,以得到所述第一计算图中的每个节点对应的算子名称、算子参数、输入数据的维度、输出数据的维度、节点标号;根据所述每个节点对应的算子名称、算子参数、输入数据的维度、输出数据的维度、节点标号生成第二计算图,其中,所述第二计算图的结构为所述芯片开发使用的计算图结构,所述第二计算图为芯片开发环境的输入。
全文数据:
权利要求:
百度查询: 深圳云天励飞技术股份有限公司 江苏云天励飞技术有限公司 对接TVM的方法及相关设备
免责声明
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。