Модуль: 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: реализация all-reduce для CrossDeviceOps с использованием NCCL.

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, если вызвана в контексте между репликами.