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

Операции, связанные с устройствами обработки тензоров.

Модули

profiler модуль: Файл-заглушка для поддержания обратной совместимости.

Классы

class AsyncCheckpointSaverHook: Сохраняет контрольные точки каждые N шагов или секунд.

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

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

class InfeedQueue: Вспомогательный объект для построения очереди ввода данных на устройстве.

class InputPipelineConfig: См. определение этих значений в TPUConfig.

class RunConfig: RunConfig с поддержкой TPU.

class TPUConfig: Конфигурация, связанная с TPU, необходимая для TPUEstimator.

class TPUDistributionStrategy: Стратегия запуска Keras-модели на TPU.

class TPUEstimator: Estimator с поддержкой TPU.

class TPUEstimatorSpec: Операции и объекты, возвращаемые из model_fn и передаваемые в TPUEstimator.

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

Функции

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

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

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

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

device_assignment(...): Вычисляет device_assignment вычисления по топологии TPU.

export_estimator_savedmodel(...): Экспорт обученной модели Estimator для инференции на TPU.

infeed_dequeue(...): Заглушка-операция для значения, которое будет подано на вычисление.

infeed_dequeue_tuple(...): Заглушка-операция для значений, подаваемых в TPU одновременно в виде кортежа.

infeed_enqueue(...): Операция, подающая единственное значение Tensor в вычисление.

infeed_enqueue_tuple(...): Подача нескольких значений Tensor в вычисление в виде кортежа XLA.

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

keras_to_tpu_model(...): Копирование model вместе с весами на TPU. (устарело)

outfeed_dequeue(...): Получает один тензор из выходного буфера вычисления.

outfeed_dequeue_tuple(...): Получение нескольких значений из выходного буфера вычисления.

outfeed_enqueue(...): Добавление Tensor в выходной буфер вычисления.

outfeed_enqueue_tuple(...): Добавление нескольких значений Tensor в выходной буфер вычисления.

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

repeat(...): Построение цикла обучения, выполняющего фиксированное количество итераций.

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

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

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

shutdown_system(...): Выключение работающей распределённой системы TPU.

while_loop(...): Построение цикла обучения для TPUs.