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
)

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