InfeedEnqueuePrelinearizedBuffer

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

Операция, которая ставит предварительно линеаризованный буфер в очередь на вход TPU.

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

сорт InfeedEnqueuePrelinearizedBuffer.Options Необязательные атрибуты для InfeedEnqueuePrelinearizedBuffer

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

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

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

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

public static InfeedEnqueuePrelinearizedBuffer create (область области действия , ввод операнда <?>, параметры... параметры)

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

Параметры
объем текущий объем
вход Вариант тензора, представляющий линеаризованный вывод.
параметры содержит значения необязательных атрибутов
Возврат
  • новый экземпляр InfeedEnqueuePrelinearizedBuffer

общедоступный статический InfeedEnqueuePrelinearizedBuffer.Options deviceOrdinal (Long deviceOrdinal)

Параметры
устройствоПорядковый номер Используемое устройство ТПУ. Это значение должно быть равно -1, когда Op работает на устройстве TPU, и = 0, когда Op работает на устройстве CPU.