Issue
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?
Solution
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);
Answered By - mrry
0 comments:
Post a Comment
Note: Only a member of this blog may post a comment.