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

摘要

本发明提供一种在TensorFlow中支持FPGA训练的方法，包括以下步骤：在TensorFlow中添加对FPGA设备的注册和发现，以使得所述FPGA设备的名称在所述TensorFlow的设备列表中；根据所述FPGA设备的名称调用TensorFlow的算子注册接口注册支持所述FPGA设备的算子，并使所述算子的名称与所述TensorFlow支持的所有设备的算子的名称相同；利用opencl编写所述算子的执行函数，包括主机端执行函数和FPGA设备端执行函数，以执行所述主机上的CPU与所述FPGA的数据交互。本发明扩大了FPGA的使用场景，适用于需要在线训练并更新模型的场景。