org.tensorflow.op.tpu

Des classes

AllToAll <T étend TType > Une opération pour échanger des données entre les répliques TPU.
CollectivePermute <T étend TType > Une opération pour permuter les tenseurs sur les instances TPU répliquées.
Résultat de la compilation Renvoie le résultat d'une compilation TPU.
CompileSucceededAssert Affirme que la compilation a réussi.
ConfigurerDistributedTPU Configure les structures centralisées pour un système TPU distribué.
ConfigurerDistributedTPU.Options Attributs facultatifs pour ConfigureDistributedTPU
ConfigurerTPUEmbedding Configure TPUEmbedding dans un système TPU distribué.
CrossReplicaSum <T étend TNumber > Une opération pour additionner les entrées sur les instances TPU répliquées.
IncorporationActivations Une opération permettant de différencier les TPU Embeddings.
Mettre en file d'attenteTPUEmbeddingIntegerBatch Une opération qui met en file d'attente une liste de tenseurs de lots d'entrée dans TPUEmbedding.
Mettre en file d'attenteTPUEmbeddingIntegerBatch.Options Attributs facultatifs pour EnqueueTPUEmbeddingIntegerBatch
Mettre en file d'attenteTPUEmbeddingRaggedTensorBatch Facilite le portage du code qui utilise tf.nn.embedding_lookup().
Mettre en file d'attenteTPUEmbeddingRaggedTensorBatch.Options Attributs facultatifs pour EnqueueTPUEmbeddingRaggedTensorBatch
Mettre en file d'attenteTPUEmbeddingSparseBatch Une opération qui met en file d'attente les indices d'entrée TPUEmbedding à partir d'un SparseTensor.
Mettre en file d'attenteTPUEmbeddingSparseBatch.Options Attributs facultatifs pour EnqueueTPUEmbeddingSparseBatch
Mettre en file d'attenteTPUEmbeddingSparseTensorBatch Facilite le portage du code qui utilise tf.nn.embedding_lookup_sparse().
Mettre en file d'attenteTPUEmbeddingSparseTensorBatch.Options Attributs facultatifs pour EnqueueTPUEmbeddingSparseTensorBatch
Exécuter Op qui charge et exécute un programme TPU sur un périphérique TPU.
Exécuter et mettre à jour les variables Op qui exécute un programme avec des mises à jour facultatives de variables sur place.
InfeedDequeue <T étend TType > Une opération d'espace réservé pour une valeur qui sera introduite dans le calcul.
InfeedDequeueTuple Récupère plusieurs valeurs de l’alimentation sous forme de tuple XLA.
EntréeEnqueue Une opération qui alimente une seule valeur Tensor dans le calcul.
InfeedEnqueue.Options Attributs facultatifs pour InfeedEnqueue
InfeedEnqueuePrelinearizedBuffer Une opération qui met en file d'attente le tampon prélinéarisé dans l'alimentation TPU.
InfeedEnqueuePrelinearizedBuffer.Options Attributs facultatifs pour InfeedEnqueuePrelinearizedBuffer
InfeedEnqueueTuple Introduire plusieurs valeurs Tensor dans le calcul sous forme de tuple XLA.
InfeedEnqueueTuple.Options Attributs facultatifs pour InfeedEnqueueTuple
LoadTPUEmbeddingAdadeltaParameters Chargez les paramètres d’intégration d’Adadelta.
LoadTPUEmbeddingAdadeltaParameters.Options Attributs facultatifs pour LoadTPUEmbeddingAdadeltaParameters
LoadTPUEmbeddingAdadeltaParametersGradAccumDebug Chargez les paramètres Adadelta avec prise en charge du débogage.
LoadTPUEmbeddingAdadeltaParametersGradAccumDebug.Options Attributs facultatifs pour LoadTPUEmbeddingAdadeltaParametersGradAccumDebug
LoadTPUEmbeddingAdagradParameters Chargez les paramètres d’intégration d’Adagrad.
LoadTPUEmbeddingAdagradParameters.Options Attributs facultatifs pour LoadTPUEmbeddingAdagradParameters
LoadTPUEmbeddingAdagradParametersGradAccumDebug Chargez les paramètres d'intégration d'Adagrad avec la prise en charge du débogage.
LoadTPUEmbeddingAdagradParametersGradAccumDebug.Options Attributs facultatifs pour LoadTPUEmbeddingAdagradParametersGradAccumDebug
LoadTPUEmbeddingADAMParameters Chargez les paramètres d’intégration ADAM.
LoadTPUEmbeddingADAMParameters.Options Attributs facultatifs pour LoadTPUEmbeddingADAMParameters
LoadTPUEmbeddingADAMParametersGradAccumDebug Chargez les paramètres d’intégration ADAM avec prise en charge du débogage.
LoadTPUEmbeddingADAMParametersGradAccumDebug.Options Attributs facultatifs pour LoadTPUEmbeddingADAMParametersGradAccumDebug
LoadTPUEmbeddingCenteredRMSPropParameters Paramètres d’intégration RMSProp centrés sur la charge.
LoadTPUEmbeddingCenteredRMSPropParameters.Options Attributs facultatifs pour LoadTPUEmbeddingCenteredRMSPropParameters
LoadTPUEmbeddingFTRLParameters Chargez les paramètres d’intégration FTRL.
LoadTPUEmbeddingFTRLParameters.Options Attributs facultatifs pour LoadTPUEmbeddingFTRLParameters
LoadTPUEmbeddingFTRLParametersGradAccumDebug Chargez les paramètres d’intégration FTRL avec prise en charge du débogage.
LoadTPUEmbeddingFTRLParametersGradAccumDebug.Options Attributs facultatifs pour LoadTPUEmbeddingFTRLParametersGradAccumDebug
LoadTPUEmbeddingMDLAdagradLightParameters Chargez les paramètres d’intégration de MDL Adagrad Light.
LoadTPUEmbeddingMDLAdagradLightParameters.Options Attributs facultatifs pour LoadTPUEmbeddingMDLAdagradLightParameters
LoadTPUEmbeddingMomentumParameters Chargez les paramètres d’intégration de Momentum.
LoadTPUEmbeddingMomentumParameters.Options Attributs facultatifs pour LoadTPUEmbeddingMomentumParameters
LoadTPUEmbeddingMomentumParametersGradAccumDebug Chargez les paramètres d’intégration de Momentum avec prise en charge du débogage.
LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options Attributs facultatifs pour LoadTPUEmbeddingMomentumParametersGradAccumDebug
LoadTPUEmbeddingProximalAdagradParameters Chargez les paramètres d’intégration proximaux d’Adagrad.
LoadTPUEmbeddingProximalAdagradParameters.Options Attributs facultatifs pour LoadTPUEmbeddingProximalAdagradParameters
LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug Chargez les paramètres d’intégration proximaux d’Adagrad avec prise en charge du débogage.
LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug.Options Attributs facultatifs pour LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug
LoadTPUEmbeddingProximalYogiParameters
LoadTPUEmbeddingProximalYogiParameters.Options Attributs facultatifs pour LoadTPUEmbeddingProximalYogiParameters
LoadTPUEmbeddingProximalYogiParametersGradAccumDebug
LoadTPUEmbeddingProximalYogiParametersGradAccumDebug.Options Attributs facultatifs pour LoadTPUEmbeddingProximalYogiParametersGradAccumDebug
LoadTPUEmbeddingRMSPropParameters Chargez les paramètres d’intégration RMSProp.
LoadTPUEmbeddingRMSPropParameters.Options Attributs facultatifs pour LoadTPUEmbeddingRMSPropParameters
LoadTPUEmbeddingRMSPropParametersGradAccumDebug Chargez les paramètres d’intégration RMSProp avec prise en charge du débogage.
LoadTPUEmbeddingRMSPropParametersGradAccumDebug.Options Attributs facultatifs pour LoadTPUEmbeddingRMSPropParametersGradAccumDebug
LoadTPUEmbeddingStochasticGradientDescentParameters Chargez les paramètres d'intégration SGD.
LoadTPUEmbeddingStochasticGradientDescentParameters.Options Attributs facultatifs pour LoadTPUEmbeddingStochasticGradientDescentParameters
LoadTPUEIncorporationStochasticGradientDescentParametersGradAccumDebug Chargez les paramètres d'intégration SGD.
LoadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug.Options Attributs facultatifs pour LoadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug
Sélecteur ordinal Un sélecteur de noyau TPU Op.
OutfeedDequeue <T étend TType > Récupère un seul tenseur de la sortie de calcul.
OutfeedDequeue.Options Attributs facultatifs pour OutfeedDequeue
OutfeedDequeueTuple Récupérez plusieurs valeurs de la sortie de calcul.
OutfeedDequeueTuple.Options Attributs facultatifs pour OutfeedDequeueTuple
SortieDequeueTupleV2 Récupérez plusieurs valeurs de la sortie de calcul.
OutfeedDequeueV2 <T étend TType > Récupère un seul tenseur de la sortie de calcul.
SortieEnqueue Mettez en file d'attente un Tensor sur la sortie de calcul.
OutfeedEnqueueTuple Mettez en file d'attente plusieurs valeurs Tensor sur la sortie de calcul.
PartitionedInput <T étend TType > Une opération qui regroupe une liste d’entrées partitionnées.
PartitionedInput.Options Attributs facultatifs pour PartitionedInput
PartitionedOutput <T étend TType > Une opération qui démultiplexe un tenseur à partitionner par XLA en une liste de partitionnés

sorties en dehors du calcul XLA.

PartitionedOutput.Options Attributs facultatifs pour PartitionedOutput
Prélinéariser Une opération qui linéarise une valeur de Tensor en un tenseur variant opaque.
Prélinéariser.Options Attributs facultatifs pour Prelinearize
PrélinéariserTuple Une opération qui linéarise plusieurs valeurs de Tensor en un tenseur variant opaque.
PrélinéariserTuple.Options Attributs facultatifs pour PrelinearizeTuple
RecvTPUEmbeddingActivations Une opération qui reçoit les activations d’intégration sur le TPU.
ReplicatedInput <T étend TType > Connecte N entrées à un calcul TPU répliqué à N voies.
ReplicatedInput.Options Attributs facultatifs pour ReplicatedInput
ReplicatedOutput <T étend TType > Connecte N sorties d’un calcul TPU répliqué à N voies.
Répliquer les métadonnées Métadonnées indiquant comment le calcul TPU doit être répliqué.
ReplicateMetadata.Options Attributs facultatifs pour ReplicateMetadata
RécupérerTPUEmbeddingAdadeltaParameters Récupérez les paramètres d’intégration d’Adadelta.
RécupérerTPUEmbeddingAdadeltaParameters.Options Attributs facultatifs pour RetrieveTPUEmbeddingAdadeltaParameters
RécupérerTPUEmbeddingAdadeltaParametersGradAccumDebug Récupérez les paramètres d’intégration d’Adadelta avec la prise en charge du débogage.
RécupérerTPUEmbeddingAdadeltaParametersGradAccumDebug.Options Attributs facultatifs pour RetrieveTPUEmbeddingAdadeltaParametersGradAccumDebug
RécupérerTPUEmbeddingAdagradParameters Récupérez les paramètres d’intégration d’Adagrad.
RécupérerTPUEmbeddingAdagradParameters.Options Attributs facultatifs pour RetrieveTPUEmbeddingAdagradParameters
RécupérerTPUEmbeddingAdagradParametersGradAccumDebug Récupérez les paramètres d’intégration d’Adagrad avec la prise en charge du débogage.
RécupérerTPUEmbeddingAdagradParametersGradAccumDebug.Options Attributs facultatifs pour RetrieveTPUEmbeddingAdagradParametersGradAccumDebug
Récupérer les paramètres TPUEmbeddingADAMParameters Récupérez les paramètres d’intégration ADAM.
RécupérerTPUEmbeddingADAMParameters.Options Attributs facultatifs pour RetrieveTPUEmbeddingADAMParameters
RécupérerTPUEmbeddingADAMParametersGradAccumDebug Récupérez les paramètres d’intégration ADAM avec la prise en charge du débogage.
RécupérerTPUEmbeddingADAMParametersGradAccumDebug.Options Attributs facultatifs pour RetrieveTPUEmbeddingADAMParametersGradAccumDebug
Récupérer les paramètres TPUEmbeddingCenteredRMSPropParameters Récupérez les paramètres d’intégration RMSProp centrés.
RécupérerTPUEmbeddingCenteredRMSPropParameters.Options Attributs facultatifs pour RetrieveTPUEmbeddingCenteredRMSPropParameters
Récupérer les paramètres TPUEmbeddingFTRLParameters Récupérez les paramètres d’intégration FTRL.
RécupérerTPUEmbeddingFTRLParameters.Options Attributs facultatifs pour RetrieveTPUEmbeddingFTRLParameters
RécupérerTPUEmbeddingFTRLParametersGradAccumDebug Récupérez les paramètres d’intégration FTRL avec la prise en charge du débogage.
RécupérerTPUEmbeddingFTRLParametersGradAccumDebug.Options Attributs facultatifs pour RetrieveTPUEmbeddingFTRLParametersGradAccumDebug
RécupérerTPUEmbeddingMDLAdagradLightParameters Récupérez les paramètres d’intégration de MDL Adagrad Light.
RécupérerTPUEmbeddingMDLAdagradLightParameters.Options Attributs facultatifs pour RetrieveTPUEmbeddingMDLAdagradLightParameters
RécupérerTPUEmbeddingMomentumParameters Récupérez les paramètres d’intégration de Momentum.
RécupérerTPUEmbeddingMomentumParameters.Options Attributs facultatifs pour RetrieveTPUEmbeddingMomentumParameters
RécupérerTPUEmbeddingMomentumParametersGradAccumDebug Récupérez les paramètres d’intégration de Momentum avec la prise en charge du débogage.
Récupérer TPUEmbeddingMomentumParametersGradAccumDebug.Options Attributs facultatifs pour RetrieveTPUEmbeddingMomentumParametersGradAccumDebug
RécupérerTPUEmbeddingProximalAdagradParameters Récupérez les paramètres d’intégration proximaux d’Adagrad.
RécupérerTPUEmbeddingProximalAdagradParameters.Options Attributs facultatifs pour RetrieveTPUEmbeddingProximalAdagradParameters
RécupérerTPUEmbeddingProximalAdagradParametersGradAccumDebug Récupérez les paramètres d’intégration proximaux d’Adagrad avec la prise en charge du débogage.
RécupérerTPUEmbeddingProximalAdagradParametersGradAccumDebug.Options Attributs facultatifs pour RetrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug
RécupérerTPUEmbeddingProximalYogiParameters
RécupérerTPUEmbeddingProximalYogiParameters.Options Attributs facultatifs pour RetrieveTPUEmbeddingProximalYogiParameters
RécupérerTPUEmbeddingProximalYogiParametersGradAccumDebug
RécupérerTPUEmbeddingProximalYogiParametersGradAccumDebug.Options Attributs facultatifs pour RetrieveTPUEmbeddingProximalYogiParametersGradAccumDebug
RécupérerTPUEmbeddingRMSPropParameters Récupérez les paramètres d’intégration RMSProp.
Récupérer TPUEmbeddingRMSPropParameters.Options Attributs facultatifs pour RetrieveTPUEmbeddingRMSPropParameters
RécupérerTPUEmbeddingRMSPropParametersGradAccumDebug Récupérez les paramètres d’intégration de RMSProp avec la prise en charge du débogage.
Récupérer TPUEmbeddingRMSPropParametersGradAccumDebug.Options Attributs facultatifs pour RetrieveTPUEmbeddingRMSPropParametersGradAccumDebug
RécupérerTPUEmbeddingStochasticGradientDescentParameters Récupérez les paramètres d’intégration SGD.
RécupérerTPUEmbeddingStochasticGradientDescentParameters.Options Attributs facultatifs pour RetrieveTPUEmbeddingStochasticGradientDescentParameters
RécupérerTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug Récupérez les paramètres d'intégration SGD avec la prise en charge du débogage.
RécupérerTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug.Options Attributs facultatifs pour RetrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug
EnvoyerTPUEmbeddingGradients Effectue des mises à jour dégradées des tables d'intégration.
ArrêtDistribuéTPU Arrête un système TPU distribué en cours d’exécution.
Résultat de compilation TPU Renvoie le résultat d'une compilation TPU.
TPUEmbeddingActivations Une opération permettant de différencier les TPU Embeddings.
TPUReplicatedInput <T étend TType > Connecte N entrées à un calcul TPU répliqué à N voies.
TPUReplicatedInput.Options Attributs facultatifs pour TPUReplicatedInput
TPUReplicatedOutput <T étend TType > Connecte N sorties d’un calcul TPU répliqué à N voies.
TPUReplicateMetadonnées Métadonnées indiquant comment le calcul TPU doit être répliqué.
TPUReplicateMetadata.Options Attributs facultatifs pour TPUReplicateMetadata
TravailleurHeartbeat Battement de coeur du travailleur op.