Модуль: tf.compat.v1.tpu

Операции, связанные с процессорами Tensor Processing Units.

Модули

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

Классы

class CrossShardOptimizer: Оптимизатор, усредняющий градиенты по фрагментам TPU.

class PaddingSpec: Представляет тип политик заполнения для tpu.replicate.

class XLAOptions: Параметры компиляции XLA.

Функции

batch_parallel(...): Разбивает фрагменты computation по размеру пакета для параллельного выполнения.

bfloat16_scope(...): Класс области видимости для переменных bfloat16, чтобы модель использовала пользовательский метод получения.

core(...): Возвращает имя устройства для ядра в реплицированном вычислении TPU.

cross_replica_sum(...): Суммирует тензор входных данных по репликам в соответствии с group_assignment.

initialize_system(...): Инициализирует распределённую систему TPU для использования с TensorFlow.

outside_compilation(...): Создаёт часть вычисления вне текущей области видимости репликации TPU.

replicate(...): Создаёт оператор графа, который выполняет реплицированное вычисление TPU.

rewrite(...): Переписывает computation для выполнения на системе TPU.

shard(...): Разбивает фрагменты computation для параллельного выполнения.

shutdown_system(...): Останавливает работающую распределённую систему TPU.