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

НЕ РЕДАКТИРОВАТЬ.

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

Классы

class DataParallel: Распределение для параллельной обработки данных.

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

class LayoutMap: Объект, похожий на словарь, который сопоставляет строку с экземплярами TensorLayout.

class ModelParallel: Распределение, которое фрагментирует переменные модели.

class TensorLayout: Макет для применения к тензору.

Функции

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

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

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

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

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