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

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

Модули

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

Классы

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

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

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

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

Функции

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

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