публичный финальный класс EnqueueTPUEmbeddingIntegerBatch
Операция, которая ставит список входных пакетных тензоров в очередь в TPUEmbedding.
Вложенные классы
| сорт | EnqueueTPUEmbeddingIntegerBatch.Options | Дополнительные атрибуты для EnqueueTPUEmbeddingIntegerBatch | |
Константы
| Нить | OP_NAME | Название этой операции, известное основному движку TensorFlow. |
Публичные методы
| статический 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 в задаче, на которой размещен узел. |
|---|