Как использовать XLA с C ++ API в тензорном потоке

Tensorflow предоставляет способ запуска графа тензорного потока через XLA, однако в нем ничего не говорится о том, как запустить граф тензорного потока через XLA с использованием API-интерфейса C++. Посоветовать кому-нибудь дать мне какой-нибудь совет?

2 ответа

Вы можете найти пошаговый пример на официальном сайте: Использование компиляции AOT

Руководство Использование компиляции AOT может быть не очень четким на каждом этапе. Ты можешь попробовать

  1. bazel build tenorflow/ компилятор /aot/tests:tfcompile_test

  2. Базэл-бен /tensorflow/ составитель / АОТ / тесты /tfcompile_test

В качестве примера.

Остальные ответы пока относятся к компиляции AOT, хотя я считаю, что JIT требуется в большинстве случаев. XLA JIT можно включить с помощью переменной окружения. Попробуйте использоватьTF_XLA_FLAGS="--tf_xla_auto_jit=1".

Другие вопросы по тегам