99网
您的当前位置:首页一种在TensorFlow中支持FPGA训练的方法和装置

一种在TensorFlow中支持FPGA训练的方法和装置

来源:99网
(19)中华人民共和国国家知识产权局

(12)发明专利申请

(21)申请号 CN201911156208.1 (22)申请日 2019.11.22

(71)申请人 苏州浪潮智能科技有限公司

地址 215100 江苏省苏州市吴中区吴中经济开发区郭巷街道官浦路1号9幢

(10)申请公布号 CN110929883A

(43)申请公布日 2020.03.27

(72)发明人 赵谦谦;仝培霖;赵红博

(74)专利代理机构 北京连和连知识产权代理有限公司

代理人 刘小峰

(51)Int.CI

权利要求说明书 说明书 幅图

(54)发明名称

一种在TensorFlow中支持FPGA训练的方法和装置

(57)摘要

本发明提供一种在TensorFlow中支持

FPGA训练的方法,包括以下步骤:在TensorFlow中添加对FPGA设备的注册和发现,以使得所述FPGA设备的名称在所述TensorFlow的设备列表中;根据所述FPGA设备的名称调用TensorFlow的算子注册接口注册支持所述FPGA设备的算子,并使所述算子的名称与所述TensorFlow支持的所有设备的算子的名称相同;利用opencl编写所述算子的执行函数,包括主机端执行函数和FPGA设

备端执行函数,以执行所述主机上的CPU与所述FPGA的数据交互。本发明扩大了FPGA的使用场景,适用于需要在线训练并更新模型的场景。

法律状态

法律状态公告日

2020-03-27 2020-03-27 2020-04-21

法律状态信息

公开 公开

实质审查的生效

法律状态

公开 公开

实质审查的生效

权利要求说明书

一种在TensorFlow中支持FPGA训练的方法和装置的权利要求说明书内容是....请下载后查看

说明书

一种在TensorFlow中支持FPGA训练的方法和装置的说明书内容是....请下载后查看

因篇幅问题不能全部显示,请点此查看更多更全内容