EnqueueTPUEmbeddingIntegerBatch

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

Операция, которая ставит список входных пакетных тензоров в очередь в TPUEmbedding.

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

сорт EnqueueTPUEmbeddingIntegerBatch.Options Дополнительные атрибуты для EnqueueTPUEmbeddingIntegerBatch

Константы

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

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

статический EnqueueTPUEmbeddingIntegerBatch
create ( Область действия, Iterable < Operand < TInt32 >> пакет, Operand < TString > modeOverride, Options... options)
Фабричный метод для создания класса, обертывающего новую операцию EnqueueTPUEmbeddingIntegerBatch.
статический EnqueueTPUEmbeddingIntegerBatch.Options
deviceOrdinal (длинный номер устройства)

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

Константы

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

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

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

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

public static EnqueueTPUEmbeddingIntegerBatch create (область области действия , Iterable < Operand < TInt32 >> пакет, Operand < TString > modeOverride, параметры ... )

Фабричный метод для создания класса, обертывающего новую операцию EnqueueTPUEmbeddingIntegerBatch.

Параметры
объем текущий объем
партия Список одномерных тензоров, по одному для каждой таблицы внедрения, содержащий индексы в таблицах.
режимOverride Строковый ввод, который переопределяет режим, указанный в TPUEmbeddingConfiguration. Поддерживаемые значения: {'unspecified', 'inference', 'training', 'backward_pass_only'}. Если установлено значение «не указано», используется режим, установленный в TPUEmbeddingConfiguration, в противном случае используется mode_override.
параметры содержит значения необязательных атрибутов
Возврат
  • новый экземпляр EnqueueTPUEmbeddingIntegerBatch

public static EnqueueTPUEmbeddingIntegerBatch.Options deviceOrdinal (Long deviceOrdinal)

Параметры
устройствоПорядковый номер Используемое устройство ТПУ. Должно быть >= 0 и меньше количества ядер TPU в задаче, на которой размещен узел.