Add New Device to TensorFlow

Question

In TensorFlow, a session places graphical operations on Devices, such as processors or GPUs, and provides methods for performing them. Can I add an additional device class to TensorFlow? Let me name XPU in addition to the processor and GPU?

Answer 1

New types of devices can be added to TensorFlow, although we do not have excellent documentation. For tensorflow::Device this will require the addition of a new tensorflow::Device implementation, adding the corresponding tensorflow::DeviceFactory and calling the following registration macro:

 REGISTER_LOCAL_DEVICE_FACTORY("XPU", MyXPUDeviceFactory);