tf.config.experimental.ClusterDeviceFilters

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

tf.config.experimental.ClusterDeviceFilters()

Примечание: это экспериментальный API и может быть изменён.

Установите фильтры устройств для выборочных задач и процессов. Для каждого удалённого рабочего узла фильтры устройств представляют собой список строк. При наличии любых фильтров удалённый рабочий игнорирует все устройства, которые не соответствуют ни одному из фильтров. Каждый фильтр может быть частично указан, например, "/job:ps", "/job:worker/replica:3" и т.д. Обратите внимание, что устройство всегда видно рабочему, на котором оно расположено.

Например, чтобы установить фильтры устройств для кластера серверов параметров:

cdf = tf.config.experimental.ClusterDeviceFilters()
for i in range(num_workers):
  cdf.set_device_filters('worker', i, ['/job:ps'])
for i in range(num_ps):
  cdf.set_device_filters('ps', i, ['/job:worker'])

tf.config.experimental_connect_to_cluster(cluster_def,
                                          cluster_device_filters=cdf)

Фильтры устройств могут быть частично указаны. Для удалённых задач, для которых не указаны фильтры устройств, все устройства будут видны им.

Методы

set_device_filters

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

set_device_filters(
    job_name, task_index, device_filters
)

Установите фильтры устройств для данного имени задачи и идентификатора.