CODEWITHSUNDEEP
tensorflowxpu
GDG
GDG

Reputation: 409

Add a new device in TensorFlow

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?

Upvotes: 6

Views: 1857

Answers (1)

mrry
mrry

Reputation: 126154

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);

Upvotes: 8

Related Questions