TensorFlow函數(shù)：tf.graph_util.must_run_on_cpu

tf.graph_util.must_run_on_cpu函數(shù)

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

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

如果給定的node_def必須在CPU上運(yùn)行,則返回True,否則返回False.

參數(shù)：

• node：要分配給設(shè)備的節(jié)點(diǎn)；可以是ops.Operation或者NodeDef.
• pin_variables_on_cpu：當(dāng)為True時(shí),如果node_def表示與變量相關(guān)的操作,則此函數(shù)將返回False.

返回值：

如果給定節(jié)點(diǎn)必須在CPU上運(yùn)行,則返回True,否則為False.