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

Публичный API пространства имен tf._api.v2.tpu

Модули

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

Классы

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

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

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

Функции

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

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

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

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

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

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

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

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

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

shutdown_system(...): завершает работу запущенной распределенной системы TPU.