tf.config.get_soft_device_placement

Возвращает состояние флага мягкого размещения устройств.

Просмотр псевдонимов

Псевдонимы совместимости для миграции

Дополнительную информацию см. в руководстве по миграции.

tf.compat.v1.config.get_soft_device_placement

tf.config.get_soft_device_placement()

Если включено, операции могут быть размещены на разных устройствах, отличных от явно назначенного пользователем устройства. Это потенциально влечет значительные затраты на производительность из-за увеличения обмена данными между устройствами.

В некоторых случаях soft_device_placement изменяет назначение устройства:

отсутствует реализация GPU/TPU для операции
нет известных или зарегистрированных устройств GPU
необходимо совместное размещение с входными данными типа ref, которые поступают с процессора
операция не может быть скомпилирована XLA. Это характерно для TPU, который всегда требует компилятора XLA.

Для TPU, если этот параметр имеет значение true, активируется функция автоматической внешней компиляции. Автоматическая внешняя компиляция переместит нескомпилируемые операции внутри программы 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