TPUReplicateMetadata

публичный финальный класс TPUReplicationMetadata

Метаданные, указывающие, как следует реплицировать вычисления TPU.

Эта операция содержит метаданные, общие для операций подграфа вычислений tpu.replication().

Вложенные классы

сорт ТПУРепликатетаметадата.Параметры Необязательные атрибуты для TPUReplicateMetadata .

Константы

Нить OP_NAME Название этой операции, известное основному движку TensorFlow.

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

статический 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)

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

Константы

общедоступная статическая финальная строка OP_NAME

Название этой операции, известное основному движку TensorFlow.

Постоянное значение: «TPUReplicateMetadata».

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

общедоступный статический 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 paddingMap (List<String> paddingMap)

общедоступный статический TPUReplicateMetadata.Options StepMarkerLocation (String StepMarkerLocation)

общедоступная статическая топология TPUReplicateMetadata.Options (топология String)

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

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

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

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