在TensorFlow中添加新设备

Question

在TensorFlow中，会话将图形操作放在 Devices 上，例如 CPU s或 GPU s ，并提供执行它们的方法。是否可以在TensorFlow中添加其他类设备？除了CPU和GPU之外，我们称之为XPU吗？

Answer 1

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

REGISTER_LOCAL_DEVICE_FACTORY("XPU", MyXPUDeviceFactory);