aggiungere un nuovo dispositivo in tensorflow

Question

In tensorflow, una sessione posti dei PO grafico sul Dispositivi, come CPU s o GPU s, e fornisce metodi per eseguire loro. È possibile aggiungere una classe aggiuntiva di dispositivi a TensorFlow? Chiamiamolo XPU oltre a CPU e GPU?

Answer 1

È possibile aggiungere nuovi tipi di dispositivo a TensorFlow, anche se non abbiamo una buona documentazione. Approssimativamente, richiederebbe l'aggiunta di una nuova implementazione di tensorflow::Device, l'aggiunta di una corrispondente implementazione di tensorflow::DeviceFactory, e invocando la seguente macro di registrazione:

REGISTER_LOCAL_DEVICE_FACTORY("XPU", MyXPUDeviceFactory);