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
)

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