codecamp

TensorFlow函数：tf.graph_util.must_run_on_cpu

tf.graph_util.must_run_on_cpu函数

tf.graph_util.must_run_on_cpu(
    node,
    pin_variables_on_cpu=False
)

定义在：tensorflow/python/framework/graph_util_impl.py.

如果给定的node_def必须在CPU上运行,则返回True,否则返回False.

参数：

  • node：要分配给设备的节点；可以是ops.Operation或者NodeDef.
  • pin_variables_on_cpu：当为True时,如果node_def表示与变量相关的操作,则此函数将返回False.

返回值：

如果给定节点必须在CPU上运行,则返回True,否则为False.

如何操作TensorFlow张量图的节点
TensorFlow函数：tf.graph_util.remove_training_nodes
温馨提示
下载编程狮App，免费阅读超1000+编程语言教程
取消
确定
目录

TensorFlow 入门基础

TensorFlow 指南

TensorFlow 函数介绍

TensorFlow 函数模块：tf

bitwise模块

TensorFlow compat模块

TensorFlow contrib模块

TensorFlow的errors模块

estimator模块

TensorFlow怎么使用gfile模块

TensorFlow的graph_util模块

TensorFlow的image模块

TensorFlow使用之tf.initializers

TensorFlow使用之tf.io

TensorFlow函数教程：tf.io.gfile

TensorFlow使用之tf.keras

TensorFlow函数教程：tf.keras.activations

TensorFlow函数教程：tf.keras.applications

TensorFlow函数教程：tf.keras.backend

TensorFlow回调函数：tf.keras.callbacks

TensorFlow使用之tf.layers

TensorFlow使用之Python/tf.linalg

TensorFlow使用之tf.logging

TensorFlow使用之tf.losses

TensorFlow使用之tf.math

TensorFlow使用之tf.metrics

TensorFlow使用之tf.manip

TensorFlow使用之tf.nn

TensorFlow使用之tf.profiler

TensorFlow使用之tf.python_io

TensorFlow使用之tf.queue

如何导出estimator

TensorFlow使用之tf.lite

TensorFlow 功能函数

TensorFlow的MNIST手写数字分类问题

关闭

 MIP.setData({ 'pageTheme' : getCookie('pageTheme') || {'day':true, 'night':false}, 'pageFontSize' : getCookie('pageFontSize') || 20 }); MIP.watch('pageTheme', function(newValue){ setCookie('pageTheme', JSON.stringify(newValue)) }); MIP.watch('pageFontSize', function(newValue){ setCookie('pageFontSize', newValue) }); function setCookie(name, value){ var days = 1; var exp = new Date(); exp.setTime(exp.getTime() + days*24*60*60*1000); document.cookie = name + '=' + value + ';expires=' + exp.toUTCString(); } function getCookie(name){ var reg = new RegExp('(^| )' + name + '=([^;]*)(;|$)'); return document.cookie.match(reg) ? JSON.parse(document.cookie.match(reg)[2]) : null; }