[发明专利]一种在TensorFlow中支持FPGA训练的方法和装置在审
申请号: | 201911156208.1 | 申请日: | 2019-11-22 |
公开(公告)号: | CN110929883A | 公开(公告)日: | 2020-03-27 |
发明(设计)人: | 赵谦谦;仝培霖;赵红博 | 申请(专利权)人: | 苏州浪潮智能科技有限公司 |
主分类号: | G06N20/00 | 分类号: | G06N20/00 |
代理公司: | 北京连和连知识产权代理有限公司 11278 | 代理人: | 刘小峰 |
地址: | 215100 江苏省苏州市吴*** | 国省代码: | 江苏;32 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | 本发明提供一种在TensorFlow中支持FPGA训练的方法,包括以下步骤:在TensorFlow中添加对FPGA设备的注册和发现,以使得所述FPGA设备的名称在所述TensorFlow的设备列表中;根据所述FPGA设备的名称调用TensorFlow的算子注册接口注册支持所述FPGA设备的算子,并使所述算子的名称与所述TensorFlow支持的所有设备的算子的名称相同;利用opencl编写所述算子的执行函数,包括主机端执行函数和FPGA设备端执行函数,以执行所述主机上的CPU与所述FPGA的数据交互。本发明扩大了FPGA的使用场景,适用于需要在线训练并更新模型的场景。 | ||
搜索关键词: | 一种 tensorflow 支持 fpga 训练 方法 装置 | ||
【主权项】:
暂无信息
下载完整专利技术内容需要扣除积分,VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于苏州浪潮智能科技有限公司,未经苏州浪潮智能科技有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/patent/201911156208.1/,转载请声明来源钻瓜专利网。