tf.compat.v1.graph_util.must_run_on_cpu
Возвращает True, если заданный узел node_def должен выполняться на процессоре, иначе 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, если заданный узел должен выполняться на процессоре, иначе 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/r2.3/api_docs/python/tf/compat/v1/graph_util/must_run_on_cpu
