Reputation: 165
Tensorflow give a way to run tensorflow graph via XLA,however, it mention nothing about how to run tensorflow graph via XLA using c++ api. Counld someone give me some advise?
Upvotes: 1
Views: 1896
Reputation: 327
The other answers so far are of AOT compilation, while I believe JIT is needed in most of times. XLA JIT can be turned on with environment variable. Try using TF_XLA_FLAGS="--tf_xla_auto_jit=1"
.
Upvotes: -1
Reputation: 221
The guide Using AOT compilation might not be very clear in each steps. You can try
bazel build tensorflow/compiler/aot/tests:tfcompile_test
bazel-bin/tensorflow/compiler/aot/tests/tfcompile_test
as an example.
Upvotes: 0