TensorFlow1.0引入了XLA支持,包括JIT编译和AOT编译。对于JIT编译,我找到了一个python测试脚本,可以使用它对其进行单元测试。但是,我还没有找到任何用于AOT编译的python测试。但也有bazel测试,可以在源码树上运行。Tensorflow的链接https://www.tensorflow.org/performance/xla/tfcompile提供了有关如何测试的信息。但是tfcompile并不包含在tensorflow的分发内容中。在这一点上我可能错了。但是我在TF的发行目录中看不到tfcompile的安装位置。谁能帮助我理解如何在现有的分发内容上测试AOT编译,或者我需要在代码中调整一些东西,以允许AOT内容进入分发?提前谢谢。
发布于 2017-03-11 06:05:25
我知道您特别想了解AOT,但我建议您首先阅读此页面:https://www.tensorflow.org/performance/xla/
然后阅读这篇文章:https://www.tensorflow.org/performance/xla/jit
特别要注意的是,我们的二进制发行版中并没有包含XLA;此时您必须从源代码构建。请注意,在运行./configure
时必须选择"enable XLA“才能启用XLA支持。
一旦您做到了这一点,Yaroslav Bulatov的建议是正确的;您可以自己构建二进制文件,或者通过bazel运行测试。
https://stackoverflow.com/questions/42719117
复制相似问题