Spec-Zone.ru › TensorFlow 2.4

tf.config.set_soft_device_placement

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

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

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

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

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

tf.compat.v1.config.set_soft_device_placement

tf.config.set_soft_device_placement(
    enabled
)

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

  1. нет реализации оператора на GPU
  2. нет известных или зарегистрированных устройств GPU
  3. необходимо разместить совместно с входом(ами) типа 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

Spec-Zone.ru

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