Spec-Zone.ru › TensorFlow 1.15

tf.config.set_soft_device_placement

Просмотреть исходный код на GitHub

Устанавливает, включена ли мягкая (soft) установка устройства.

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

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

См. Руководство по миграции для получения дополнительных сведений.

tf.compat.v1.config.set_soft_device_placement, `tf.compat.v2.config.set_soft_device_placement`

tf.config.set_soft_device_placement(
    enabled
)

Если включено, операция будет размещена на CPU, если выполняется хотя бы одно из следующих условий:

  1. для данной операции нет реализации на GPU
  2. неизвестны или не зарегистрированы устройства GPU
  3. необходимо размещение совместно с входными данными типа ref, которые находятся на CPU
Аргументы
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/r1.15/api_docs/python/tf/config/set_soft_device_placement

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API