TPUOrdinalSelector

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

Селектор ядра ТПУ Op.

Эта операция создает набор ядер TPU (для разминки) или одно ядро ​​TPU (для обычного вывода) для выполнения программы TPU. Выходные данные используются TPUPartitionedCall.

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

Вывод <Целое число>
какВывод ()
Возвращает символический дескриптор тензора.
статический TPUOrdinalSelector
создать (область действия )
Фабричный метод для создания класса, обертывающего новую операцию TPUOrdinalSelector.
Вывод <Целое число>
Ординалы устройства ()
Вектор из 1 или более ядер ТПУ.

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

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

общедоступный вывод <Integer> asOutput ()

Возвращает символический дескриптор тензора.

Входные данные для операций TensorFlow являются выходными данными другой операции TensorFlow. Этот метод используется для получения символического дескриптора, который представляет собой вычисление входных данных.

public static TPUOrdinalSelector create (область области действия )

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

Параметры
объем текущий объем
Возврат
  • новый экземпляр TPUOrdinalSelector

общедоступный вывод <Integer> deviceOrdinals ()

Вектор из 1 или более ядер ТПУ.