tensorflow XLA 笔记

笔记

通过session_config 开启XLA
特定op关闭 XLA

通过session_config 开启XLA

from tensorflow.core.protobuf import config_pb2
from tensorflow.core.protobuf import rewriter_config_pb2
import tensorflow as tfsession_config = tf.ConfigProto(allow_soft_placement=True,log_device_placement=False)# 开启XLA
session_config.graph_options.optimizer_options.global_jit_level =  config_pb2.OptimizerOptions.ON_1# 或者开启XLA默认的优化
# session_config.graph_options.optimizer_options.global_jit_level =  config_pb2.OptimizerOptions.DEFAULT# 关闭XLA
# session_config.graph_options.optimizer_options.global_jit_level = config_pb2.OptimizerOptions.OFFconfig=tf.estimator.RunConfig(session_config=session_config,log_step_count_steps=1,protocol="grpc",keep_checkpoint_max=5,save_checkpoints_steps=10)estimator = tf.estimator.Estimator(model_fn=model_fn,  #用户定义的model_fnmodel_dir=model_dir,  # 保存ckpt 的路径config=config,)

特定op关闭 XLA

from tensorflow.python.compiler.xla import jit
with jit.experimental_jit_scope(compile_ops=False):a = tf.add(1, 2)

tensorflow XLA 笔记相关推荐

TensorFlow XLA优化与Memory
TensorFlow XLA优化与Memory XLA概述 XLA(加速线性代数)是用于优化TensorFlow计算的线性代数的域特定编译器.结果是在服务器与移动平台上的速度,内存使用率与可移植性得到 ...
TensorFlow XLA加速编译器
TensorFlow XLA加速编译器加速线性代数器(Accelerated linear algebra,XLA)是线性代数领域的专用编译器.根据 https://www.tensorflow.o ...
Tensorflow学习笔记6：解决tensorflow训练过程中GPU未调用问题
Tensorflow学习笔记6:解决tensorflow训练过程中GPU未调用问题参考文章: (1)Tensorflow学习笔记6:解决tensorflow训练过程中GPU未调用问题 (2)http ...
tensorflow学习笔记(三十二):conv2d_transpose (解卷积)
tensorflow学习笔记(三十二):conv2d_transpose ("解卷积") deconv解卷积,实际是叫做conv_transpose, conv_transpose ...
tensorflow学习笔记二——建立一个简单的神经网络拟合二次函数
tensorflow学习笔记二--建立一个简单的神经网络 2016-09-23 16:04 2973人阅读评论(2) 收藏举报分类: tensorflow(4) 目录(?)[+] 本笔记目的 ...
tensorflow学习笔记——使用TensorFlow操作MNIST数据（1）
续集请点击我:tensorflow学习笔记--使用TensorFlow操作MNIST数据(2) 本节开始学习使用tensorflow教程,当然从最简单的MNIST开始.这怎么说呢,就好比编程入门有He ...
TensorFlow学习笔记（二）：快速理解Tutorial第一个例子-MNIST机器学习入门标签：机器学习SoftmaxTensorFlow教程 2016-08-02 22:12 3729人阅
TensorFlow学习笔记(二):快速理解Tutorial第一个例子-MNIST机器学习入门标签: 机器学习SoftmaxTensorFlow教程 2016-08-02 22:12 3729人阅读 ...
Tensorflow学习笔记2：About Session, Graph, Operation and Tensor
简介上一篇笔记:Tensorflow学习笔记1:Get Started 我们谈到Tensorflow是基于图(Graph)的计算系统.而图的节点则是由操作(Operation)来构成的,而图的各个节 ...
Win10：tensorflow学习笔记（4）
前言学以致用,以学促用.输出检验,完整闭环. 经过前段时间的努力,已经在电脑上搭好了深度学习系统,接下来就要开始跑程序了,将AI落地了. 安装win10下tensforlow 可以参照之前的例子:w ...

tensorflow XLA 笔记

笔记

通过session_config 开启XLA

特定op关闭 XLA

tensorflow XLA 笔记相关推荐

最新文章

热门文章