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