tf.config.set_soft_device_placement
| Просмотреть исходный код на GitHub |
Устанавливает, включена ли мягкая размещение устройств.
tf.config.set_soft_device_placement(
enabled
)
Если включено, операция будет размещена на процессоре, если выполняются какие-либо из следующих условий
- нет реализации оператора на GPU
- нет известных или зарегистрированных устройств GPU
- необходимо разместить совместно с входом(ами) типа ref, которые получены с процессора
| Аргументы | |
|---|---|
enabled | Включить ли мягкое размещение. |
© 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.4/api_docs/python/tf/config/set_soft_device_placement