org.tensorflow.op.tpu

Классы

AllToAll <T расширяет TType > Операция для обмена данными между репликами TPU.
CollectivePermute <T расширяет TType > Операция для перестановки тензоров в реплицируемых экземплярах TPU.
Результат Компиляции Возвращает результат компиляции TPU.
КомпиляцияУспехАссерт Утверждает, что компиляция прошла успешно.
КонфигурераспределенныйТПУ Устанавливает централизованные структуры для распределенной системы TPU.
КонфигуредистрибутедТПУ.Параметры Дополнительные атрибуты для ConfigureDistributedTPU
НастройкаTPUEmbedding Настраивает TPUEmbedding в распределенной системе TPU.
CrossReplicaSum <T расширяет TNumber > Операция для суммирования входных данных по реплицируемым экземплярам TPU.
Встраивание Активаций Операция, позволяющая дифференцировать встраивания TPU.
EnqueueTPUEmbeddingIntegerBatch Операция, которая ставит список входных пакетных тензоров в очередь в TPUEmbedding.
EnqueueTPUEmbeddingIntegerBatch.Options Дополнительные атрибуты для EnqueueTPUEmbeddingIntegerBatch
EnqueueTPUEmbeddingRaggedTensorBatch Упрощает перенос кода, использующего tf.nn.embedding_lookup().
EnqueueTPUEmbeddingRaggedTensorBatch.Options Необязательные атрибуты для EnqueueTPUEmbeddingRaggedTensorBatch
EnqueueTPUEmbeddingSparseBatch Операция, которая ставит в очередь входные индексы TPUEmbedding из SparseTensor.
EnqueueTPUEmbeddingSparseBatch.Options Необязательные атрибуты для EnqueueTPUEmbeddingSparseBatch
EnqueueTPUEmbeddingSparseTensorBatch Упрощает перенос кода, использующего tf.nn.embedding_lookup_sparse().
EnqueueTPUEmbeddingSparseTensorBatch.Options Необязательные атрибуты для EnqueueTPUEmbeddingSparseTensorBatch
Выполнять Операция, которая загружает и выполняет программу TPU на устройстве TPU.
Выполнение и обновление переменных Операция, выполняющая программу с необязательными обновлениями переменных на месте.
InfeedDequeue <T расширяет TType > Заполнитель для значения, которое будет использоваться в вычислениях.
InfeedDequeueTuple Извлекает несколько значений из ввода в виде кортежа XLA.
InfeedEnqueue Операция, которая вводит в вычисления одно значение Тензора.
InfeedEnqueue.Options Необязательные атрибуты для InfeedEnqueue
InfeedEnqueuePrelinearizedBuffer Операция, которая ставит предварительно линеаризованный буфер в очередь на вход TPU.
InfeedEnqueuePrelinearizedBuffer.Options Необязательные атрибуты для InfeedEnqueuePrelinearizedBuffer
InfeedEnqueueTuple Подает несколько значений Tensor в вычисления в виде кортежа XLA.
InfeedEnqueueTuple.Options Необязательные атрибуты для InfeedEnqueueTuple
ЗагрузитьTPUEmbeddingAdadeltaParameters Загрузите параметры внедрения Adadelta.
LoadTPUEmbeddingAdadeltaParameters.Options Дополнительные атрибуты для LoadTPUEmbeddingAdadeltaParameters
LoadTPUEmbeddingAdadeltaParametersGradAccumDebug Загрузка параметров Adadelta с поддержкой отладки.
LoadTPUEmbeddingAdadeltaParametersGradAccumDebug.Options Дополнительные атрибуты для LoadTPUEmbeddingAdadeltaParametersGradAccumDebug
ЗагрузитьTPUEmbeddingAdagradParameters Загрузите параметры внедрения Adagrad.
LoadTPUEmbeddingAdagradParameters.Options Дополнительные атрибуты для LoadTPUEmbeddingAdagradParameters
LoadTPUEmbeddingAdagradParametersGradAccumDebug Загрузка параметров внедрения Adagrad с поддержкой отладки.
LoadTPUEmbeddingAdagradParametersGradAccumDebug.Options Дополнительные атрибуты для LoadTPUEmbeddingAdagradParametersGradAccumDebug
Загрузить параметры TPUEmbeddingADAMP Загрузите параметры внедрения ADAM.
ЗагрузитьTPUEmbeddingADAMParameters.Options Дополнительные атрибуты для LoadTPUEmbeddingADAMParameters
ЗагрузитьTPUEmbeddingADAMParametersGradAccumDebug Загрузка параметров внедрения ADAM с поддержкой отладки.
LoadTPUEmbeddingADAMParametersGradAccumDebug.Options Дополнительные атрибуты для LoadTPUEmbeddingADAMParametersGradAccumDebug
Лоадтпуембеддингцентредрсмпроппараметерс Параметры внедрения RMSProp, ориентированные на нагрузку.
LoadTPUEmbeddingCenteredRMSPropParameters.Options Дополнительные атрибуты для LoadTPUEmbeddingCenteredRMSPropParameters
ЗагрузитьTPUEmbeddingFTRLParameters Загрузите параметры внедрения FTRL.
LoadTPUEmbeddingFTRLParameters.Options Дополнительные атрибуты для LoadTPUEmbeddingFTRLParameters .
ЗагрузитьTPUEmbeddingFTRLParametersGradAccumDebug Загрузка параметров внедрения FTRL с поддержкой отладки.
LoadTPUEmbeddingFTRLParametersGradAccumDebug.Options Дополнительные атрибуты для LoadTPUEmbeddingFTRLParametersGradAccumDebug
LoadTPUEmbeddingMDLAdagradLightParameters Загрузите параметры встраивания MDL Adagrad Light.
LoadTPUEmbeddingMDLAdagradLightParameters.Options Дополнительные атрибуты для LoadTPUEmbeddingMDLAdagradLightParameters
LoadTPUEmbeddingMomentumParameters Загрузите параметры встраивания Momentum.
LoadTPUEmbeddingMomentumParameters.Options Дополнительные атрибуты для LoadTPUEmbeddingMomentumParameters
LoadTPUEmbeddingMomentumParametersGradAccumDebug Загрузка параметров внедрения Momentum с поддержкой отладки.
LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options Дополнительные атрибуты для LoadTPUEmbeddingMomentumParametersGradAccumDebug
LoadTPUEmbeddingProximalAdagradParameters Загрузите проксимальные параметры встраивания Adagrad.
LoadTPUEmbeddingProximalAdagradParameters.Options Дополнительные атрибуты для LoadTPUEmbeddingProximalAdagradParameters
LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug Загрузка проксимальных параметров внедрения Adagrad с поддержкой отладки.
LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug.Options Дополнительные атрибуты для LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug
LoadTPUEmbeddingProximalYogiParameters
LoadTPUEmbeddingProximalYogiParameters.Options Дополнительные атрибуты для LoadTPUEmbeddingProximalYogiParameters
LoadTPUEmbeddingProximalYogiParametersGradAccumDebug
LoadTPUEmbeddingProximalYogiParametersGradAccumDebug.Options Дополнительные атрибуты для LoadTPUEmbeddingProximalYogiParametersGradAccumDebug
ЗагрузитьTPUEmbeddingRMSPropParameters Загрузите параметры внедрения RMSProp.
LoadTPUEmbeddingRMSPropParameters.Options Дополнительные атрибуты для LoadTPUEmbeddingRMSPropParameters
LoadTPUEmbeddingRMSPropParametersGradAccumDebug Загрузите параметры внедрения RMSProp с поддержкой отладки.
LoadTPUEmbeddingRMSPropParametersGradAccumDebug.Options Дополнительные атрибуты для LoadTPUEmbeddingRMSPropParametersGradAccumDebug
LoadTPUEmbeddingStochasticGradientDescentParameters Загрузите параметры внедрения SGD.
LoadTPUEmbeddingStochasticGradientDescentParameters.Options Дополнительные атрибуты для LoadTPUEmbeddingStochasticGradientDescentParameters
LoadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug Загрузите параметры внедрения SGD.
LoadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug.Options Дополнительные атрибуты для LoadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug
Порядковый селектор Селектор ядра ТПУ Op.
OutfeedDequeue <T расширяет TType > Извлекает один тензор из исходящих вычислений.
OutfeedDequeue.Options Необязательные атрибуты для OutfeedDequeue
OutfeedDequeueTuple Получите несколько значений из исходящего потока вычислений.
OutfeedDequeueTuple.Options Необязательные атрибуты для OutfeedDequeueTuple
OutfeedDequeueTupleV2 Получите несколько значений из исходящего потока вычислений.
OutfeedDequeueV2 <T расширяет TType > Извлекает один тензор из исходящих вычислений.
OutfeedEnqueue Поставьте тензор в очередь на выходе вычислений.
OutfeedEnqueueTuple Поставьте в очередь несколько значений Tensor на выходе вычислений.
PartitionedInput <T расширяет TType > Операция, которая группирует список разделенных входных данных.
PartitionedInput.Параметры Дополнительные атрибуты для PartitionedInput
PartitionedOutput <T расширяет TType > Операция, которая демультиплексирует тензор, который будет сегментирован XLA в список секционированных

выходные данные вне вычислений XLA.

PartitionedOutput.Options Дополнительные атрибуты для PartitionedOutput
Предварительная линеаризация Операция, которая линеаризует одно значение тензора в непрозрачный вариант тензора.
Предварительная линеаризация.Параметры Дополнительные атрибуты для Prelinearize
Предварительная линеаризация кортежа Операция, которая линеаризует несколько значений тензора в непрозрачный вариант тензора.
PrelinearizeTuple.Options Необязательные атрибуты для PrelinearizeTuple
RecvTPUEmbeddingActivations Операция, которая получает активации встраивания в TPU.
РеплицированныйИнпут <T расширяет TType > Подключает N входов к N-образному реплицированному вычислению TPU.
РеплицированныйИнпут.Параметры Необязательные атрибуты для ReplicatedInput
ReplicationOutput <T расширяет TType > Соединяет N выходов N-путевого реплицированного вычисления TPU.
Реплицировать метаданные Метаданные, указывающие, как следует реплицировать вычисления TPU.
РеплицироватьМетаданные.Параметры Необязательные атрибуты для ReplicateMetadata
Получить параметры TPUEmbeddingAdadeltaParameters Получите параметры внедрения Adadelta.
ПолучитьTPUEmbeddingAdadeltaParameters.Options Необязательные атрибуты для RetrieveTPUEmbeddingAdadeltaParameters
ПолучитьTPUEEmbeddingAdadeltaParametersGradAccumDebug Получение параметров внедрения Adadelta с поддержкой отладки.
ПолучитьTPUEmbeddingAdadeltaParametersGradAccumDebug.Options Необязательные атрибуты для RetrieveTPUEmbeddingAdadeltaParametersGradAccumDebug
Получить параметры TPUEmbeddingAdagrad Получить параметры внедрения Adagrad.
ПолучитьTPUEmbeddingAdagradParameters.Options Необязательные атрибуты для RetrieveTPUEmbeddingAdagradParameters
ПолучитьTPUEembeddingAdagradParametersGradAccumDebug Получение параметров внедрения Adagrad с поддержкой отладки.
ПолучитьTPUEmbeddingAdagradParametersGradAccumDebug.Options Необязательные атрибуты для RetrieveTPUEmbeddingAdagradParametersGradAccumDebug
Получение параметров TPUEmbeddingADAMP Получите параметры внедрения ADAM.
ПолучитьTPUEmbeddingADAMParameters.Options Необязательные атрибуты для RetrieveTPUEmbeddingADAMParameters
ПолучитьTPUEmbeddingADAMParametersGradAccumDebug Получение параметров внедрения ADAM с поддержкой отладки.
ПолучитьTPUEmbeddingADAMParametersGradAccumDebug.Options Необязательные атрибуты для RetrieveTPUEmbeddingADAMParametersGradAccumDebug
ПолучитьTPUEmbeddingCenteredRMSPropParameters Получите центрированные параметры внедрения RMSProp.
ПолучитьTPUEmbeddingCenteredRMSPropParameters.Options Необязательные атрибуты для RetrieveTPUEmbeddingCenteredRMSPropParameters
Получить параметры TPUEmbeddingFTRLParameters Получите параметры внедрения FTRL.
ПолучитьTPUEmbeddingFTRLParameters.Options Необязательные атрибуты для RetrieveTPUEmbeddingFTRLParameters
ПолучитьTPUEmbeddingFTRLParametersGradAccumDebug Получение параметров внедрения FTRL с поддержкой отладки.
ПолучитьTPUEmbeddingFTRLParametersGradAccumDebug.Options Необязательные атрибуты для RetrieveTPUEmbeddingFTRLParametersGradAccumDebug
Получить параметры TPUEmbeddingMDLAdagradLight Получить параметры встраивания MDL Adagrad Light.
ПолучитьTPUEmbeddingMDLAdagradLightParameters.Options Необязательные атрибуты для RetrieveTPUEmbeddingMDLAdagradLightParameters
Получить параметры TPUEmbeddingMomentum Получить параметры встраивания Momentum.
ПолучитьTPUEmbeddingMomentumParameters.Options Необязательные атрибуты для RetrieveTPUEmbeddingMomentumParameters
ПолучитьTPUEmbeddingMomentumParametersGradAccumDebug Получите параметры внедрения Momentum с поддержкой отладки.
ПолучитьTPUEmbeddingMomentumParametersGradAccumDebug.Options Необязательные атрибуты для RetrieveTPUEmbeddingMomentumParametersGradAccumDebug
Получить параметры TPUEmbeddingProximalAdagradParameters Получите проксимальные параметры встраивания Adagrad.
ПолучитьTPUEmbeddingProximalAdagradParameters.Options Необязательные атрибуты для RetrieveTPUEmbeddingProximalAdagradParameters
ПолучитьTPUEmbeddingProximalAdagradParametersGradAccumDebug Получение проксимальных параметров внедрения Adagrad с поддержкой отладки.
ПолучитьTPUEmbeddingProximalAdagradParametersGradAccumDebug.Options Необязательные атрибуты для RetrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug
ПолучитьTPUEmbeddingProximalYogiParameters
ПолучитьTPUEmbeddingProximalYogiParameters.Options Необязательные атрибуты для RetrieveTPUEmbeddingProximalYogiParameters
ПолучитьTPUEEmbeddingProximalYogiParametersGradAccumDebug
ПолучитьTPUEmbeddingProximalYogiParametersGradAccumDebug.Options Необязательные атрибуты для RetrieveTPUEmbeddingProximalYogiParametersGradAccumDebug
ПолучитьTPUEmbeddingRMSPropParameters Получите параметры внедрения RMSProp.
ПолучитьTPUEmbeddingRMSPropParameters.Options Необязательные атрибуты для RetrieveTPUEmbeddingRMSPropParameters
ПолучитьTPUEmbeddingRMSPropParametersGradAccumDebug Получение параметров внедрения RMSProp с поддержкой отладки.
ПолучитьTPUEmbeddingRMSPropParametersGradAccumDebug.Options Необязательные атрибуты для RetrieveTPUEmbeddingRMSPropParametersGradAccumDebug
ПолучитьTPUEmbeddingStochasticGradientDescentParameters Получите параметры внедрения SGD.
ПолучитьTPUEmbeddingStochasticGradientDescentParameters.Options Необязательные атрибуты для RetrieveTPUEmbeddingStochasticGradientDescentParameters
ПолучитьTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug Получите параметры внедрения SGD с поддержкой отладки.
ПолучитьTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug.Options Необязательные атрибуты для RetrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug
ОтправитьТПУЭмбеддингГрадиентс Выполняет градиентное обновление таблиц внедрения.
ВыключениеРаспределенныйТПУ Выключает работающую распределенную систему TPU.
ТПУКомпиляцияРезультат Возвращает результат компиляции TPU.
TPUEmbeddingActivations Операция, позволяющая дифференцировать встраивания TPU.
TPUReplicationInput <T расширяет TType > Подключает N входов к N-образному реплицированному вычислению TPU.
ТПУРеплицедИнпут.Параметры Дополнительные атрибуты для TPUReplicatedInput
TPUReplicationOutput <T расширяет TType > Соединяет N выходов N-путевого реплицированного вычисления TPU.
ТПУреплицироватьметаданные Метаданные, указывающие, как следует реплицировать вычисления TPU.
ТПУРепликатетаметадата.Параметры Необязательные атрибуты для TPUReplicateMetadata .
РабочийСердцебиение Сердцебиение рабочего оп.