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
)

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