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)

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

Из класса org.tensorflow.op.RawOp

последнее логическое значение	равно (Объект obj)
окончательный интервал	хэш-код ()
Операция	оп () Верните эту единицу вычислений как одну Operation .
последняя строка	нанизывать ()

Из класса java.lang.Object

логическое значение	равно (Объект arg0)
последний класс<?>	получитьКласс ()
интервал	хэш-код ()
окончательная пустота	поставить в известность ()
окончательная пустота	уведомитьВсе ()
Нить	нанизывать ()
окончательная пустота	подождать (длинный arg0, int arg1)
окончательная пустота	подождите (длинный arg0)
окончательная пустота	ждать ()

Из интерфейса org.tensorflow.op.Op

абстрактная среда выполнения	окружение () Верните среду выполнения, в которой была создана эта операция.
абстрактная операция	оп () Верните эту единицу вычислений как одну Operation .

Константы

общедоступная статическая финальная строка 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.