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

Публичный API для пространства имён tf.tpu.experimental.

Модули

embedding модуль: Публичный API для пространства имён tf.tpu.experimental.embedding.

Классы

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

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

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

Функции

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

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