ReplicateMetadata.Options

общедоступный статический класс ReplicationMetadata.Options

Необязательные атрибуты для ReplicateMetadata

Публичные методы

РеплицироватьМетаданные.Параметры
allowSoftPlacement (логическое значениеallowSoftPlacement)
РеплицироватьМетаданные.Параметры
ComputingShape (List<Long> ComputingShape)
РеплицироватьМетаданные.Параметры
deviceAssignment (List<Long> deviceAssignment)
РеплицироватьМетаданные.Параметры
хостComputeCore (Список<String> хостComputeCore)
РеплицироватьМетаданные.Параметры
numCoresPerReplica (длинное numCoresPerReplica)
РеплицироватьМетаданные.Параметры
paddingMap (Список<String> paddingMap)
РеплицироватьМетаданные.Параметры
StepMarkerLocation (Строка StepMarkerLocation)
РеплицироватьМетаданные.Параметры
топология (строковая топология)
РеплицироватьМетаданные.Параметры
useSpmdForXlaPartitioning (логическое значение useSpmdForXlaPartitioning)
РеплицироватьМетаданные.Параметры
useTpu (логическое значение useTpu)

Унаследованные методы

Публичные методы

public ReplicationMetadata.OptionsallowSoftPlacement ( логическое значениеallowSoftPlacement)

public ReplicationMetadata.Options ComputingShape (List<Long> ComputingShape)

Параметры
ВычислениеФорма УСТАРЕЛО. Вместо этого используйте num_cores_per_replica.

public ReplicationMetadata.Options deviceAssignment (List<Long> deviceAssignment)

Параметры
устройствоНазначение Назначение устройств для вычислений.

общедоступный ReplicationMetadata.Options хостComputeCore (List<String> хостComputeCore)

public ReplicationMetadata.Options numCoresPerReplica (длинное numCoresPerReplica)

Параметры
numCoresPerReplica Количество ядер на реплику. Используется для параллелизма моделей.

public ReplicationMetadata.Options paddingMap (List<String> paddingMap)

public ReplicationMetadata.Options StepMarkerLocation (String StepMarkerLocation)

общедоступная топология ReplicationMetadata.Options (строковая топология)

Параметры
топология TopologyProto, указывающий топологию среза модуля TPU.

public ReplicationMetadata.Options useSpmdForXlaPartitioning (логическое значение useSpmdForXlaPartitioning)

public ReplicationMetadata.Options useTpu (логическое значение useTpu)

Параметры
использовать ТПУ Размещать ли вычисления на TPU.