在TensorFlow

Question

添加新设备在TensorFlow，一个会话地方的图表OPS到设备，如CPU S或GPU s和提供方法来执行它们。是否可以向TensorFlow添加额外的设备类？除了CPU和GPU之外，我们称之为XPU？

Answer 1

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

REGISTER_LOCAL_DEVICE_FACTORY("XPU", MyXPUDeviceFactory);