tf.graph_util.must_run_on_cpu
Возвращает True, если указанный узел node_def должен выполняться на процессоре CPU, в противном случае False. (устаревший)
tf.graph_util.must_run_on_cpu(
node, pin_variables_on_cpu=False
)
| Аргументы | |
|---|---|
node | Узел, который должен быть назначен устройству. Может быть либо ops.Operation, либо NodeDef. |
pin_variables_on_cpu | Если True, эта функция вернёт False, если node_def представляет операцию, связанную с переменной. |
| Возвращаемое значение | |
|---|---|
| True, если указанный узел должен выполняться на CPU, в противном случае False. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/graph_util/must_run_on_cpu
