Как использовать XLA с C ++ API в тензорном потоке
Tensorflow предоставляет способ запуска графа тензорного потока через XLA, однако в нем ничего не говорится о том, как запустить граф тензорного потока через XLA с использованием API-интерфейса C++. Посоветовать кому-нибудь дать мне какой-нибудь совет?
2 ответа
Вы можете найти пошаговый пример на официальном сайте: Использование компиляции AOT
Руководство Использование компиляции AOT может быть не очень четким на каждом этапе. Ты можешь попробовать
bazel build tenorflow/ компилятор /aot/tests:tfcompile_test
Базэл-бен /tensorflow/ составитель / АОТ / тесты /tfcompile_test
В качестве примера.
Остальные ответы пока относятся к компиляции AOT, хотя я считаю, что JIT требуется в большинстве случаев. XLA JIT можно включить с помощью переменной окружения. Попробуйте использоватьTF_XLA_FLAGS="--tf_xla_auto_jit=1"
.