Добавить новую платформу в tenorflow - PullRequest
0 голосов
/ 01 февраля 2019

Я пытаюсь создать новую платформу / устройство для тензорного потока.

После этого:
https://www.tensorflow.org/xla/developing_new_backend

Я создал StreamExecutor, а также класс Platform (похоже, он реализует все функции передачи и выполнения для устройства).
Я добавил распечатки ко всем функциям-членам Платформы для отладки / трассировки.
Я вижу во время 'импортирования тензорного потока', что моя Платформа регистрируется.
Я ожидал увидеть вызов VisibleDeviceCount () для системыдля регистрации устройств.
Я изменил VisibleDeviceCount так, чтобы он всегда возвращал 1 (т. е. одно видимое устройство).
Однако я не вижу вызовов ни VisibleDeviceCount (), ни устройства моей платформы в списке с tf.device_lib.list_local_devices ().

Есть ли какое-либо руководство или список шагов, необходимых для создания нового устройства?

...