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.