tf.compat.v1.graph_util.must_run_on_cpu
Возвращает True, если указанный узел node_def должен выполняться на процессоре CPU, в противном случае False. (устарело)
tf.compat.v1.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. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/compat/v1/graph_util/must_run_on_cpu
