Module: 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.