Модуль: tf.tpu.experimental

Общедоступный API для пространства имен tf._api.v2.tpu.experimental

Модули

embedding модуль: Общедоступный API для пространства имен tf._api.v2.tpu.experimental.embedding

Классы

class DeviceAssignment: Сопоставление логических ядер в вычислении с физической топологией TPU.

class DeviceOrderMode: Способ определения порядка устройств при вычислении задания устройства.

class HardwareFeature: Класс содержит всю информацию о функциях TPU.

class TPUSystemMetadata: Описывает некоторые метаданные о системе TPU.

class Topology: Описывает набор устройств TPU.

Функции

initialize_tpu_system(...): Инициализация устройств TPU.

shutdown_tpu_system(...): Выключение устройств TPU.