在 TensorFlow 中添加新设备

【问题标题】：Add a new device in TensorFlow在 TensorFlow 中添加新设备
【发布时间】：2016-02-04 22:47:16
【问题描述】：

在 TensorFlow 中，Session 将图形操作放置在设备上，例如 CPU 或 GPU ，并提供执行它们的方法。是否可以向 TensorFlow 添加额外的设备类别？除了CPU和GPU，我们还叫它XPU？

【问题讨论】：

【解决方案1】：

虽然我们没有很好的文档，但可以向 TensorFlow 添加新的设备类型。大约需要添加tensorflow::Device 的新实现，添加tensorflow::DeviceFactory 的相应实现，并调用以下注册宏：

REGISTER_LOCAL_DEVICE_FACTORY("XPU", MyXPUDeviceFactory);

【讨论】：