EnqueueTPUEmbeddingIntegerBatch

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

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

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

сорт

EnqueueTPUEmbeddingIntegerBatch.Options

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

Константы

Нить

OP_NAME

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

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

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

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

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

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

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

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

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

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

Константы

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