Add a new device in TensorFlow

Question

In TensorFlow, a Session places the graph ops onto Devices, such as CPUs or GPUs, and provides methods to execute them. Is it possible to add an additional class of devices to TensorFlow? Let's call it XPU in addition to CPU and GPU?

Answer 1

It is possible to add new device types to TensorFlow, although we don't have great documentation. Approximately, it would require adding a new implementation of tensorflow::Device, adding a corresponding implementation of tensorflow::DeviceFactory, and invoking the following registration macro:

REGISTER_LOCAL_DEVICE_FACTORY("XPU", MyXPUDeviceFactory);