Добавление нового устройства в TensorFlow

Question

В TensorFlow, A сессии помещает граф OPS на устройств, таких как CPU ов или GPU с, и предоставляет методы для выполнения их. Можно ли добавить дополнительный класс устройств в TensorFlow? Назовем его XPU в дополнение к процессору и графическому процессору?

Answer 1

В TensorFlow можно добавлять новые типы устройств, хотя у нас нет отличной документации. Примерно, это потребовало бы добавить новую реализацию tensorflow::Device, добавив соответствующую реализацию tensorflow::DeviceFactory, и вызов следующей регистрации макрос:

REGISTER_LOCAL_DEVICE_FACTORY("XPU", MyXPUDeviceFactory);