org.tensorflow.op.tpu

Aulas

AllToAll <T estende TType > Uma op para trocar dados entre réplicas de TPU.
CollectivePermute <T estende TType > Um Op para permutar tensores em instâncias replicadas de TPU.
CompilationResult Retorna o resultado de uma compilação de TPU.
CompileSucceededAssert Afirma que a compilação foi bem-sucedida.
ConfigureDistributedTPU Configura as estruturas centralizadas para um sistema TPU distribuído.
ConfigureDistributedTPU.Options Atributos opcionais para ConfigureDistributedTPU
ConfigureTPUEmbedding Configura o TPUEmbedding em um sistema TPU distribuído.
CrossReplicaSum <T estende TNumber > Um Op para somar entradas em instâncias de TPU replicadas.
EmbeddingActivations Uma operação que permite a diferenciação de embeddings TPU.
EnqueueTPUEmbeddingIntegerBatch Um op que enfileira uma lista de tensores de lote de entrada para TPUEmbedding.
EnqueueTPUEmbeddingIntegerBatch.Options Atributos opcionais para EnqueueTPUEmbeddingIntegerBatch
EnqueueTPUEmbeddingRaggedTensorBatch Facilita a portabilidade de código que usa tf.nn.embedding_lookup ().
EnqueueTPUEmbeddingRaggedTensorBatch.Options Atributos opcionais para EnqueueTPUEmbeddingRaggedTensorBatch
EnqueueTPUEmbeddingSparseBatch Um op que enfileira índices de entrada TPUEmbedding de um SparseTensor.
EnqueueTPUEmbeddingSparseBatch.Options Atributos opcionais para EnqueueTPUEmbeddingSparseBatch
EnqueueTPUEmbeddingSparseTensorBatch Facilita a transferência de código que usa tf.nn.embedding_lookup_sparse ().
EnqueueTPUEmbeddingSparseTensorBatch.Options Atributos opcionais para EnqueueTPUEmbeddingSparseTensorBatch
Executar Op que carrega e executa um programa TPU em um dispositivo TPU.
ExecuteAndUpdateVariables Op que executa um programa com atualizações opcionais de variáveis ​​no local.
InfeedDequeue <T estende TType > Um placeholder op para um valor que será inserido no cálculo.
InfeedDequeueTuple Busca vários valores de alimentação como uma tupla XLA.
InfeedEnqueue Uma operação que alimenta um único valor do tensor na computação.
InfeedEnqueue.Options Atributos opcionais para InfeedEnqueue
InfeedEnqueuePrelinearizedBuffer Um op que enfileira o buffer pré-linearizado na alimentação de TPU.
InfeedEnqueuePrelinearizedBuffer.Options Atributos opcionais para InfeedEnqueuePrelinearizedBuffer
InfeedEnqueueTuple Alimenta vários valores do Tensor no cálculo como uma tupla XLA.
InfeedEnqueueTuple.Options Atributos opcionais para InfeedEnqueueTuple
LoadTPUEmbeddingAdadeltaParameters Carregue os parâmetros de incorporação do Adadelta.
LoadTPUEmbeddingAdadeltaParameters.Options Atributos opcionais para LoadTPUEmbeddingAdadeltaParameters
LoadTPUEmbeddingAdadeltaParametersGradAccumDebug Carregue os parâmetros do Adadelta com suporte para depuração.
LoadTPUEmbeddingAdadeltaParametersGradAccumDebug.Options Atributos opcionais para LoadTPUEmbeddingAdadeltaParametersGradAccumDebug
LoadTPUEmbeddingAdagradParameters Carregue os parâmetros de incorporação do Adagrad.
LoadTPUEmbeddingAdagradParameters.Options Atributos opcionais para LoadTPUEmbeddingAdagradParameters
LoadTPUEmbeddingAdagradParametersGradAccumDebug Carregue os parâmetros de incorporação do Adagrad com suporte para depuração.
LoadTPUEmbeddingAdagradParametersGradAccumDebug.Options Atributos opcionais para LoadTPUEmbeddingAdagradParametersGradAccumDebug
LoadTPUEmbeddingADAMParameters Carregue os parâmetros de incorporação do ADAM.
LoadTPUEmbeddingADAMParameters.Options Atributos opcionais para LoadTPUEmbeddingADAMParameters
LoadTPUEmbeddingADAMParametersGradAccumDebug Carregue os parâmetros de incorporação do ADAM com suporte para depuração.
LoadTPUEmbeddingADAMParametersGradAccumDebug.Options Atributos opcionais para LoadTPUEmbeddingADAMParametersGradAccumDebug
LoadTPUEmbeddingCenteredRMSPropParameters Carregar parâmetros de incorporação RMSProp centralizados.
LoadTPUEmbeddingCenteredRMSPropParameters.Options Atributos opcionais para LoadTPUEmbeddingCenteredRMSPropParameters
LoadTPUEmbeddingFTRLParameters Carregue os parâmetros de incorporação FTRL.
LoadTPUEmbeddingFTRLParameters.Options Atributos opcionais para LoadTPUEmbeddingFTRLParameters
LoadTPUEmbeddingFTRLParametersGradAccumDebug Carregar parâmetros de incorporação FTRL com suporte a depuração.
LoadTPUEmbeddingFTRLParametersGradAccumDebug.Options Atributos opcionais para LoadTPUEmbeddingFTRLParametersGradAccumDebug
LoadTPUEmbeddingMDLAdagradLightParameters Carregar parâmetros de incorporação MDL Adagrad Light.
LoadTPUEmbeddingMDLAdagradLightParameters.Options Atributos opcionais para LoadTPUEmbeddingMDLAdagradLightParameters
LoadTPUEmbeddingMomentumParameters Carregue os parâmetros de incorporação do Momentum.
LoadTPUEmbeddingMomentumParameters.Options Atributos opcionais para LoadTPUEmbeddingMomentumParameters
LoadTPUEmbeddingMomentumParametersGradAccumDebug Carregue os parâmetros de incorporação do Momentum com suporte para depuração.
LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options Atributos opcionais para LoadTPUEmbeddingMomentumParametersGradAccumDebug
LoadTPUEmbeddingProximalAdagradParameters Carregue os parâmetros de incorporação Adagrad proximais.
LoadTPUEmbeddingProximalAdagradParameters.Options Atributos opcionais para LoadTPUEmbeddingProximalAdagradParameters
LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug Carregue parâmetros de incorporação Adagrad proximais com suporte para depuração.
LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug.Options Atributos opcionais para LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug
LoadTPUEmbeddingProximalYogiParameters
LoadTPUEmbeddingProximalYogiParameters.Options Atributos opcionais para LoadTPUEmbeddingProximalYogiParameters
LoadTPUEmbeddingProximalYogiParametersGradAccumDebug
LoadTPUEmbeddingProximalYogiParametersGradAccumDebug.Options Atributos opcionais para LoadTPUEmbeddingProximalYogiParametersGradAccumDebug
LoadTPUEmbeddingRMSPropParameters Carregue os parâmetros de incorporação RMSProp.
LoadTPUEmbeddingRMSPropParameters.Options Atributos opcionais para LoadTPUEmbeddingRMSPropParameters
LoadTPUEmbeddingRMSPropParametersGradAccumDebug Carregar parâmetros de incorporação RMSProp com suporte para depuração.
LoadTPUEmbeddingRMSPropParametersGradAccumDebug.Options Atributos opcionais para LoadTPUEmbeddingRMSPropParametersGradAccumDebug
LoadTPUEmbeddingStochasticGradientDescentParameters Carregue os parâmetros de incorporação SGD.
LoadTPUEmbeddingStochasticGradientDescentParameters.Options Atributos opcionais para LoadTPUEmbeddingStochasticGradientDescentParameters
LoadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug Carregue os parâmetros de incorporação SGD.
LoadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug.Options Atributos opcionais para LoadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug
OrdinalSelector Um seletor de núcleo TPU Op.
OutfeedDequeue <T estende TType > Recupera um único tensor do outfeed de computação.
OutfeedDequeue.Options Atributos opcionais para OutfeedDequeue
OutfeedDequeueTuple Recupere vários valores do outfeed de computação.
OutfeedDequeueTuple.Options Atributos opcionais para OutfeedDequeueTuple
OutfeedDequeueTupleV2 Recupere vários valores do outfeed de computação.
OutfeedDequeueV2 <T estende TType > Recupera um único tensor do outfeed de computação.
OutfeedEnqueue Enfileire um tensor no outfeed de computação.
OutfeedEnqueueTuple Enfileirar vários valores do Tensor no outfeed de computação.
PartitionedInput <T estende TType > Uma operação que agrupa uma lista de entradas particionadas.
PartitionedInput.Options Atributos opcionais para PartitionedInput
PartitionedOutput <T estende TType > Uma operação que demultiplexa um tensor a ser fragmentado por XLA em uma lista de particionados

saídas fora da computação XLA.

PartitionedOutput.Options Atributos opcionais para PartitionedOutput
Prelinearizar Um op que lineariza um valor do tensor para um tensor variante opaco.
Prelinearize.Options Atributos opcionais para Prelinearize
PrelinearizeTuple Um op que lineariza vários valores do tensor em um tensor variante opaco.
PrelinearizeTuple.Options Atributos opcionais para PrelinearizeTuple
RecvTPUEmbeddingActivations Uma operação que recebe ativações de incorporação na TPU.
ReplicatedInput <T estende TType > Conecta N entradas a uma computação TPU replicada de N vias.
ReplicatedInput.Options Atributos opcionais para ReplicatedInput
ReplicatedOutput <T estende TType > Conecta N saídas de uma computação TPU replicada N-way.
ReplicateMetadata Metadados que indicam como a computação da TPU deve ser replicada.
ReplicateMetadata.Options Atributos opcionais para ReplicateMetadata
RetrieveTPUEmbeddingAdadeltaParameters Recupere os parâmetros de incorporação do Adadelta.
RetrieveTPUEmbeddingAdadeltaParameters.Options Atributos opcionais para RetrieveTPUEmbeddingAdadeltaParameters
RetrieveTPUEmbeddingAdadeltaParametersGradAccumDebug Recupere os parâmetros de incorporação do Adadelta com suporte para depuração.
RetrieveTPUEmbeddingAdadeltaParametersGradAccumDebug.Options Atributos opcionais para RetrieveTPUEmbeddingAdadeltaParametersGradAccumDebug
RetrieveTPUEmbeddingAdagradParameters Recupere parâmetros de incorporação Adagrad.
RetrieveTPUEmbeddingAdagradParameters.Options Atributos opcionais para RetrieveTPUEmbeddingAdagradParameters
RetrieveTPUEmbeddingAdagradParametersGradAccumDebug Recupere parâmetros de incorporação Adagrad com suporte para depuração.
RetrieveTPUEmbeddingAdagradParametersGradAccumDebug.Options Atributos opcionais para RetrieveTPUEmbeddingAdagradParametersGradAccumDebug
RetrieveTPUEmbeddingADAMParameters Recupere os parâmetros de incorporação do ADAM.
RetrieveTPUEmbeddingADAMParameters.Options Atributos opcionais para RetrieveTPUEmbeddingADAMParameters
RetrieveTPUEmbeddingADAMParametersGradAccumDebug Recupere os parâmetros de incorporação do ADAM com suporte para depuração.
RetrieveTPUEmbeddingADAMParametersGradAccumDebug.Options Atributos opcionais para RetrieveTPUEmbeddingADAMParametersGradAccumDebug
RetrieveTPUEmbeddingCenteredRMSPropParameters Recupere parâmetros de incorporação RMSProp centralizados.
RetrieveTPUEmbeddingCenteredRMSPropParameters.Options Atributos opcionais para RetrieveTPUEmbeddingCenteredRMSPropParameters
RetrieveTPUEmbeddingFTRLParameters Recupere os parâmetros de incorporação FTRL.
RetrieveTPUEmbeddingFTRLParameters.Options Atributos opcionais para RetrieveTPUEmbeddingFTRLParameters
RetrieveTPUEmbeddingFTRLParametersGradAccumDebug Recupere parâmetros de incorporação FTRL com suporte para depuração.
RetrieveTPUEmbeddingFTRLParametersGradAccumDebug.Options Atributos opcionais para RetrieveTPUEmbeddingFTRLParametersGradAccumDebug
RetrieveTPUEmbeddingMDLAdagradLightParameters Recupere parâmetros de incorporação MDL Adagrad Light.
RetrieveTPUEmbeddingMDLAdagradLightParameters.Options Atributos opcionais para RetrieveTPUEmbeddingMDLAdagradLightParameters
RetrieveTPUEmbeddingMomentumParameters Recupere os parâmetros de incorporação do Momentum.
RetrieveTPUEmbeddingMomentumParameters.Options Atributos opcionais para RetrieveTPUEmbeddingMomentumParameters
RetrieveTPUEmbeddingMomentumParametersGradAccumDebug Recupere os parâmetros de incorporação do Momentum com suporte para depuração.
RetrieveTPUEmbeddingMomentumParametersGradAccumDebug.Options Atributos opcionais para RetrieveTPUEmbeddingMomentumParametersGradAccumDebug
RetrieveTPUEmbeddingProximalAdagradParameters Recupere parâmetros de incorporação Adagrad proximais.
RetrieveTPUEmbeddingProximalAdagradParameters.Options Atributos opcionais para RetrieveTPUEmbeddingProximalAdagradParameters
RetrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug Recupere parâmetros de incorporação Adagrad proximais com suporte a depuração.
RetrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug.Options Atributos opcionais para RetrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug
RetrieveTPUEmbeddingProximalYogiParameters
RetrieveTPUEmbeddingProximalYogiParameters.Options Atributos opcionais para RetrieveTPUEmbeddingProximalYogiParameters
RetrieveTPUEmbeddingProximalYogiParametersGradAccumDebug
RetrieveTPUEmbeddingProximalYogiParametersGradAccumDebug.Options Atributos opcionais para RetrieveTPUEmbeddingProximalYogiParametersGradAccumDebug
RetrieveTPUEmbeddingRMSPropParameters Recupere os parâmetros de incorporação RMSProp.
RetrieveTPUEmbeddingRMSPropParameters.Options Atributos opcionais para RetrieveTPUEmbeddingRMSPropParameters
RetrieveTPUEmbeddingRMSPropParametersGradAccumDebug Recupere parâmetros de incorporação RMSProp com suporte para depuração.
RetrieveTPUEmbeddingRMSPropParametersGradAccumDebug.Options Atributos opcionais para RetrieveTPUEmbeddingRMSPropParametersGradAccumDebug
RetrieveTPUEmbeddingStochasticGradientDescentParameters Recupere os parâmetros de incorporação SGD.
RetrieveTPUEmbeddingStochasticGradientDescentParameters.Options Atributos opcionais para RetrieveTPUEmbeddingStochasticGradientDescentParameters
RetrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug Recupere parâmetros de incorporação SGD com suporte a depuração.
RetrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug.Options Atributos opcionais para RetrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug
SendTPUEmbeddingGradients Executa atualizações de gradiente de tabelas de incorporação.
ShutdownDistributedTPU Encerra um sistema TPU distribuído em execução.
TPUCompilationResult Retorna o resultado de uma compilação de TPU.
TPUEmbeddingActivations Uma operação que permite a diferenciação de embeddings TPU.
TPUReplicatedInput <T estende TType > Conecta N entradas a uma computação TPU replicada de N vias.
TPUReplicatedInput.Options Atributos opcionais para TPUReplicatedInput
TPUReplicatedOutput <T estende TType > Conecta N saídas de uma computação TPU replicada N-way.
TPUReplicateMetadata Metadados que indicam como a computação da TPU deve ser replicada.
TPUReplicateMetadata.Options Atributos opcionais para TPUReplicateMetadata
WorkerHeartbeat Batimento cardíaco do trabalhador op.