публичный финальный класс TPUReplicationMetadata
Метаданные, указывающие, как следует реплицировать вычисления TPU.
Эта операция содержит метаданные, общие для операций подграфа вычислений tpu.replication().
Вложенные классы
| сорт | ТПУРепликатетаметадата.Параметры | Необязательные атрибуты для TPUReplicateMetadata . | |
Публичные методы
| статический TPUReplicateMetadata.Options | allowSoftPlacement (логическое значениеallowSoftPlacement) |
| статический TPUReplicateMetadata.Options | ComputingShape (List<Long> ComputingShape) |
| статические TPUReplicationMetadata | create (Область действия , Длинные числа реплик, Параметры... параметры) Фабричный метод для создания класса, обертывающего новую операцию TPUReplicateMetadata. |
| статический TPUReplicateMetadata.Options | deviceAssignment (List<Long> deviceAssignment) |
| статический TPUReplicateMetadata.Options | хостComputeCore (Список<String> хостComputeCore) |
| статический TPUReplicateMetadata.Options | numCoresPerReplica (длинное numCoresPerReplica) |
| статический TPUReplicateMetadata.Options | paddingMap (Список<String> paddingMap) |
| статический TPUReplicateMetadata.Options | StepMarkerLocation (Строка StepMarkerLocation) |
| статический TPUReplicateMetadata.Options | топология (строковая топология) |
| статический TPUReplicateMetadata.Options | useSpmdForXlaPartitioning (логическое значение useSpmdForXlaPartitioning) |
| статический TPUReplicateMetadata.Options | useTpu (логическое значение useTpu) |
Унаследованные методы
Публичные методы
общедоступный статический TPUReplicateMetadata.OptionsallowSoftPlacement ( BooleanallowSoftPlacement)
общедоступный статический TPUReplicateMetadata.Options ComputingShape (List<Long> ComputingShape)
Параметры
| ВычислениеФорма | УСТАРЕЛО. Вместо этого используйте num_cores_per_replica. |
|---|
public static TPUReplicateMetadata create ( Область действия, Long numReplicas, Опции... параметры)
Фабричный метод для создания класса, обертывающего новую операцию TPUReplicateMetadata.
Параметры
| объем | текущий объем |
|---|---|
| количество реплик | Количество реплик вычислений |
| параметры | содержит значения необязательных атрибутов |
Возврат
- новый экземпляр TPUReplicationMetadata
общедоступный статический TPUReplicateMetadata.Options deviceAssignment (List<Long> deviceAssignment)
Параметры
| устройствоНазначение | Назначение устройств для вычислений. |
|---|
общедоступный статический TPUReplicateMetadata.Options хостComputeCore (List<String> hostComputeCore)
общедоступный статический TPUReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)
Параметры
| numCoresPerReplica | Количество ядер на реплику. Используется для параллелизма моделей. |
|---|
общедоступный статический TPUReplicateMetadata.Options StepMarkerLocation (String StepMarkerLocation)
общедоступная статическая топология TPUReplicateMetadata.Options (топология String)
Параметры
| топология | TopologyProto, указывающий топологию среза модуля TPU. |
|---|
общедоступный статический TPUReplicateMetadata.Options useSpmdForXlaPartitioning (логическое значение useSpmdForXlaPartitioning)
общедоступный статический TPUReplicateMetadata.Options useTpu (логическое значение useTpu)
Параметры
| использовать ТПУ | Размещать ли вычисления на TPU. |
|---|