Spec-Zone.ru › TensorFlow 1.15

tf.distribute.cluster_resolver.SimpleClusterResolver

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

Простое реализация ClusterResolver, принимающая ClusterSpec.

Наследуется от: ClusterResolver

Просмотр псевдонимов

Основные псевдонимы

`tf.contrib.cluster_resolver.SimpleClusterResolver`

Псевдонимы совместимости для миграции

См. Руководство по миграции для получения дополнительных сведений.

tf.compat.v1.distribute.cluster_resolver.SimpleClusterResolver, `tf.compat.v2.distribute.cluster_resolver.SimpleClusterResolver`

tf.distribute.cluster_resolver.SimpleClusterResolver(
    cluster_spec, master='', task_type=None, task_id=None, environment='',
    num_accelerators=None, rpc_layer=None
)
Атрибуты
environment Возвращает текущую среду, в которой выполняется TensorFlow.

Возможны два значения возврата: «google» (когда TensorFlow выполняется во внутренней среде Google) или пустая строка (когда TensorFlow выполняется где-либо еще).

Если вы реализуете ClusterResolver, который работает как в среде Google, так и в открытом мире (например, TPU ClusterResolver или аналогичный), вам необходимо вернуть соответствующую строку в зависимости от среды, которую вам нужно определить.

В противном случае, если вы реализуете ClusterResolver, который будет работать только в открытом TensorFlow, вам не нужно реализовывать это свойство.

rpc_layer
task_id
task_type

Методы

cluster_spec

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

cluster_spec()

Возвращает ClusterSpec, переданный в конструктор.

master

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

master(
    task_type=None, task_id=None, rpc_layer=None
)

Возвращает адрес узла для использования при создании сессии.

Аргументы
task_type (Необязательно) Тип задачи TensorFlow узла.
task_id (Необязательно) Индекс задачи TensorFlow узла.
rpc_layer (Необязательно) RPC, используемый распределенным TensorFlow.
Возвращаемые значения
Имя или URL узла сессии.

Если заданы task_type и task_id, это переопределит строку master , переданную в функцию инициализации.

num_accelerators

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

num_accelerators(
    task_type=None, task_id=None, config_proto=None
)

Возвращает количество ускорительных ядер на рабочий узел.

SimpleClusterResolver не выполняет автоматическое обнаружение ускорителей, поэтому сессия TensorFlow никогда не будет создана, и все аргументы не используются, и мы просто предполагаем, что тип ускорителя — GPU, и возвращаем значение, переданное нам в конструкторе.

Аргументы
task_type Не используется.
task_id Не используется.
config_proto Не используется.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/distribute/cluster_resolver/SimpleClusterResolver

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API