Модуль: tf.keras.distribution

НЕ РЕДАКТИРУЙТЕ.

Этот файл сгенерирован автоматически. Не редактируйте его вручную, поскольку ваши изменения будут перезаписаны.

Классы

class DataParallel: Распределение для параллелизма по данным.

class DeviceMesh: Кластер вычислительных устройств для распределённых вычислений.

class LayoutMap: Словареподобный объект, сопоставляющий строки экземплярам TensorLayout.

class ModelParallel: Распределение, разделяющее переменные модели на сегменты.

class TensorLayout: Размещение, применяемое к тензору.

Функции

distribute_tensor(...): Изменить размещение значения Tensor во время выполнения jit-функции.

distribution(...): Получить текущее распределение из глобального контекста.

initialize(...): Инициализировать систему распределения для работы с несколькими хостами/процессами.

list_devices(...): Вернуть все доступные устройства указанного типа.

set_distribution(...): Задать распределение в качестве глобальной настройки.