публичный финальный класс 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. |
---|