Модуль: tf.compat.v1.config.experimental

Публичный API для пространства имён tf._api.v2.config.experimental

Классы

class ClusterDeviceFilters: Представляет коллекцию фильтров устройств для удалённых рабочих узлов кластера.

class VirtualDeviceConfiguration: Класс конфигурации для логических устройств.

Функции

disable_mlir_bridge(...): Отключает экспериментальный компиляторный мост TensorFlow на основе MLIR.

enable_mlir_bridge(...): Включает экспериментальный компиляторный мост TensorFlow на основе MLIR.

enable_tensor_float_32_execution(...): Включает или отключает использование TensorFloat-32 на поддерживаемом оборудовании.

get_device_details(...): Возвращает сведения о физическом устройстве.

get_device_policy(...): Получает текущую политику устройства.

get_memory_growth(...): Проверяет, включён ли рост объёма памяти для PhysicalDevice.

get_memory_info(...): Возвращает сведения об использовании памяти выбранным устройством в виде словаря.

get_memory_usage(...): Возвращает текущее использование памяти выбранным устройством в байтах. (устарело)

get_synchronous_execution(...): Проверяет, выполняются ли операции синхронно или асинхронно.

get_virtual_device_configuration(...): Возвращает конфигурацию виртуального устройства для tf.config.PhysicalDevice.

get_visible_devices(...): Возвращает список видимых физических устройств.

list_logical_devices(...): Возвращает список логических устройств, созданных средой выполнения.

list_physical_devices(...): Возвращает список физических устройств, видимых среде выполнения хоста.

reset_memory_stats(...): Сбрасывает отслеживаемую статистику памяти для выбранного устройства.

set_device_policy(...): Устанавливает политику устройства для текущего потока.

set_memory_growth(...): Задаёт, следует ли включить рост объёма памяти для PhysicalDevice.

set_synchronous_execution(...): Задаёт, должны ли операции выполняться синхронно или асинхронно.

set_virtual_device_configuration(...): Задаёт конфигурацию логического устройства для tf.config.PhysicalDevice.

set_visible_devices(...): Задаёт список видимых устройств.

tensor_float_32_execution_enabled(...): Возвращает сведения о том, включён ли TensorFloat-32.