有人在https://www.tensorflow.org/versions/master/experimental/xla/jit的第3步成功尝试了mnist_softmax_xla的示例我无法使其正常工作。我得到了:
TF_XLA_FLAGS=--xla_generate_hlo_graph=.* python mnist_softmax_xla.py
Extracting /tmp/tensorflow/mnist/input_data/train-images-idx3-ubyte.gz
Extracting /tmp/tensorflow/mnist/input_data/train-labels-idx1-ubyte.gz
Extracting /tmp/tensorflow/mnist/input_data/t10k-images-idx3-ubyte.gz
Extracting /tmp/tensorflow/mnist/input_data/t10k-labels-idx1-ubyte.gz
I tensorflow/compiler/xla/service/platform_util.cc:58] platform Host present
with 24 visible devices
I tensorflow/compiler/xla/service/service.cc:180] XLA service executing
computations on platform Host. Devices:
I tensorflow/compiler/xla/service/service.cc:187] StreamExecutor device
(0): <undefined>, <undefined>
0.9198
我在启用XLA的情况下从源代码构建TensorFlow r1.0。有人有任何想法吗?谢谢!