如何在张量流中使用xla和c ++ api

Question

Tensorflow提供了一种通过XLA运行张量流图的方法，但是，它没有提到如何使用c ++ api通过XLA运行张量流图。 有人给我一些建议吗？

Answer 1

您可以在官方网站上找到分步示例： Using AOT compilation

Answer 2

指南Using AOT compilation在每个步骤中可能都不是很清楚。 你可以尝试

1. bazel build tensorflow / compiler / aot / tests：tfcompile_test

2. 巴泽勒滨/ tensorflow /编译器/ AOT /测试/ tfcompile_test

作为一个例子。

Answer 3

到目前为止，其他答案是AOT编译，而我相信大多数时候都需要JIT。可以使用环境变量打开XLA JIT。尝试使用TF_XLA_FLAGS="--tf_xla_auto_jit=1"。