tf.config.get_soft_device_placement
Возвращает состояние флага размещения устройств в мягком режиме.
tf.config.get_soft_device_placement()
Если параметр включен, операции могут размещаться на устройствах, отличных от явно назначенного пользователем устройства. Это может значительно снизить производительность из-за увеличения объема обмена данными между устройствами.
В некоторых случаях soft_device_placement изменяет назначение устройства:
- для операции нет реализации для GPU/TPU
- не обнаружено или не зарегистрировано ни одного устройства GPU
- необходимо разместить вместе с входными данными типа reftype, поступающими с CPU
- операцию нельзя скомпилировать с помощью XLA. Это часто встречается на TPU, для которого всегда требуется компилятор XLA.
Для TPU, если этот параметр включен, активируется функция под названием «автоматическая внешняя компиляция». Автоматическая внешняя компиляция перемещает операции, которые не удается скомпилировать в программе TPU, для выполнения на хосте. Это может помочь при сбоях компиляции, вызванных неподдерживаемыми операциями.
| Возвращаемое значение | |
|---|---|
| Логическое значение, указывающее, включено ли размещение в мягком режиме. |
© 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/config/get_soft_device_placement
