TensorFlow函数:tf.graph_util.must_run_on_cpu
2018-05-16 10:46 更新
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.