1

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

4

3 回答 3

0

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

于 2020-05-20T06:36:55.337 回答
0

您可以在官方网站上找到分步示例： 使用 AOT 编译

于 2017-06-01T11:29:17.387 回答
0

使用 AOT 编译的指南在每个步骤中可能不是很清楚。你可以试试

  1. bazel 构建 tensorflow/compiler/aot/tests:tfcompile_test

  2. bazel-bin/tensorflow/编译器/aot/tests/tfcompile_test

举个例子。

于 2017-09-11T08:57:38.433 回答