(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训练的方法和装置的说明书内容是....请下载后查看