Модуль: tf.compat.v1.distribute

Общедоступный API для пространства имён tf._api.v2.distribute

Модули

cluster_resolver модуль: Общедоступный API для пространства имён tf._api.v2.distribute.cluster_resolver

experimental модуль: Общедоступный API для пространства имён tf._api.v2.distribute.experimental

Классы

class CrossDeviceOps: Базовый класс для межпроцессорного уменьшения и трансляции алгоритмов.

class HierarchicalCopyAllReduce: Реализация иерархического копирования all-reduce для CrossDeviceOps.

class InputContext: Класс, оборачивающий информацию, необходимую для функции ввода.

class InputReplicationMode: Режим репликации для функции ввода.

class MirroredStrategy: Синхронное обучение по нескольким репликам на одном компьютере.

class NcclAllReduce: Реализация NCCL all-reduce для CrossDeviceOps.

class OneDeviceStrategy: Стратегия распределения для работы на одном устройстве.

class ReduceOp: Указывает, как следует уменьшить набор значений.

class ReductionToOneDevice: Реализация CrossDeviceOps, которая копирует значения на одно устройство для уменьшения.

class ReplicaContext: Класс с набором API, которые можно вызывать в контексте реплики.

class RunOptions: Параметры запуска для strategy.run.

class Server: Внутрипроцессорный сервер TensorFlow, используемый в распределенном обучении.

class Strategy: Список устройств с политикой распределения состояния и вычислений.

class StrategyExtended: Дополнительные API для алгоритмов, которые должны быть чувствительны к распределению.

Функции

experimental_set_strategy(...): Установить tf.distribute.Strategy как текущий без with strategy.scope().

get_loss_reduction(...): tf.distribute.ReduceOp, соответствующий последнему уменьшению потерь.

get_replica_context(...): Возвращает текущий tf.distribute.ReplicaContext или None.

get_strategy(...): Возвращает текущий объект tf.distribute.Strategy.

has_strategy(...): Возвращает, есть ли текущая не по умолчанию tf.distribute.Strategy.

in_cross_replica_context(...): Возвращает True, если в контексте между репликами.