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

Общедоступный API для пространства имён tf.config.experimental.

Классы

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

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

Функции

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

disable_mlir_graph_optimization(...): Отключает экспериментальные оптимизации компилятора TensorFlow на базе MLIR.

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

enable_mlir_graph_optimization(...): Включает экспериментальные оптимизации компилятора 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.