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/api_docs/python/tf/compat/v1/graph_util/must_run_on_cpu