ein neues Gerät in TensorFlow hinzufügen

Question

In TensorFlow, eine Session Orte der Graph ops auf Devices, wie CPU s oder GPU s und bietet Methoden, um sie auszuführen. Ist es möglich, TensorFlow eine zusätzliche Klasse von Geräten hinzuzufügen? Nennen wir es neben CPU und GPU auch XPU?

Answer 1

Es ist möglich, TensorFlow neue Gerätetypen hinzuzufügen, obwohl wir keine großartige Dokumentation haben. Ungefähr würde es erfordern eine neue Implementierung von tensorflow::Device Hinzufügen Hinzufügen einer entsprechenden Implementierung von tensorflow::DeviceFactory und Aufrufen der folgenden Registrierungs Makro:

REGISTER_LOCAL_DEVICE_FACTORY("XPU", MyXPUDeviceFactory);