问从tf.slim启用XLA
EN

Stack Overflow用户

提问于 2017-04-28 22:18:51

回答 1查看 215关注 0票数 1

打开Tensorflow XLA的一种方法是使用tf.OptimizerOptions.ON_1标志，将它传递给TF会话，类似于python中的以下行：

config = tf.ConfigProto()
config.graph_options.optimizer_options.global_jit_level = tf.OptimizerOptions.ON_1  
sess = tf.Session(config=config)

但是，当我使用Slim库时，我不确定何时/如何启用XLA (我使用slim.learning.train启动培训)。

回答已采纳

发布于 2017-05-01 19:53:26

slim.learning.train采用一个session_config参数，您可以将ConfigProto传递给该参数。

票数 3

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/43689633

复制

相似问题

问从tf.slim启用XLAEN