публичный финальный класс EnqueueTPUEmbeddingIntegerBatch
Операция, которая ставит список входных пакетных тензоров в очередь в TPUEmbedding.
Вложенные классы
| сорт | EnqueueTPUEmbeddingIntegerBatch.Options | Дополнительные атрибуты для EnqueueTPUEmbeddingIntegerBatch | |
Публичные методы
| статический EnqueueTPUEmbeddingIntegerBatch | create (область области действия , пакет Iterable< Operand <Integer>>, Operand <String> modeOverride, параметры... ) Фабричный метод для создания класса, обертывающего новую операцию EnqueueTPUEmbeddingIntegerBatch. |
| статический EnqueueTPUEmbeddingIntegerBatch.Options | deviceOrdinal (длинный номер устройства) |
Унаследованные методы
Публичные методы
public static EnqueueTPUEmbeddingIntegerBatch create (область действия , пакет Iterable< Operand <Integer>>, Operand <String> modeOverride, Options... options)
Фабричный метод для создания класса, обертывающего новую операцию EnqueueTPUEmbeddingIntegerBatch.
Параметры
| объем | текущий объем |
|---|---|
| партия | Список одномерных тензоров, по одному для каждой таблицы внедрения, содержащий индексы в таблицах. |
| режимOverride | Строковый ввод, который переопределяет режим, указанный в TPUEmbeddingConfiguration. Поддерживаемые значения: {'unspecified', 'inference', 'training', 'backward_pass_only'}. Если установлено значение «не указано», используется режим, установленный в TPUEmbeddingConfiguration, в противном случае используется mode_override. |
| параметры | содержит значения необязательных атрибутов |
Возврат
- новый экземпляр EnqueueTPUEmbeddingIntegerBatch
public static EnqueueTPUEmbeddingIntegerBatch.Options deviceOrdinal (Long deviceOrdinal)
Параметры
| устройствоПорядковый номер | Используемое устройство ТПУ. Должно быть >= 0 и меньше количества ядер TPU в задаче, на которой размещен узел. |
|---|