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

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

Классы

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

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

Функции

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

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

enable_op_determinism(...): Настраивает детерминированное выполнение операций TensorFlow.

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.