在 TensorFlow 中添加新设备

Question

在 TensorFlow 中，Session 将图形操作放到 Devices 上，例如 CPUs 或 GPUs，并提供执行它们的方法。是否可以向 TensorFlow 添加额外的 class 设备？除了 CPU 和 GPU?

我们还称它为 XPU 吗？

Answer 1

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

REGISTER_LOCAL_DEVICE_FACTORY("XPU", MyXPUDeviceFactory);