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
)

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