TPUReplicateMetadata.Options

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

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

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

ТПУРепликатетаметадата.Параметры
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 TPUReplicationMetadata.OptionsallowSoftPlacement ( логическое значениеallowSoftPlacement)

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

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

общедоступный TPUReplicationMetadata.Options deviceAssignment (List<Long> deviceAssignment)

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

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

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

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

public TPUReplicateMetadata.Options paddingMap (List<String> papdingMap)

public TPUReplicateMetadata.Options StepMarkerLocation (String StepMarkerLocation)

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

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

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

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

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