TPURoundRobin

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

Циклическая балансировка нагрузки на ядрах TPU.

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

Эта операция выполняет циклический перебор между целыми числами в [0, NumTPUCoresVisiblePerHost]. Это полезно для взаимодействия с операциями TensorFlow, которые принимают в качестве входных данных ядро ​​TPU, на котором выполняются вычисления, например `TPUPartitionedCall`.

device_ordinal: целое число в [0, NumTPUCoresVisiblePerHost].

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

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

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

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

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

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

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

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

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

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

публичный вывод <Integer> deviceOrdinal ()