SendTPUEmbeddingGradients

classe finale publique SendTPUEmbeddingGradients

Effectue des mises à jour dégradées des tables d'intégration.

Constantes

Chaîne OP_NAME Le nom de cette opération, tel que connu par le moteur principal TensorFlow

Méthodes publiques

SendTPUEmbeddingGradients statique
créer ( Scope scope, Iterable< Operand < TFloat32 >> entrées, Iterable< Operand < TFloat32 >> learningRates, String config)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération SendTPUEmbeddingGradients.

Méthodes héritées

Constantes

chaîne finale statique publique OP_NAME

Le nom de cette opération, tel que connu par le moteur principal TensorFlow

Valeur constante : "SendTPUEmbeddingGradients"

Méthodes publiques

public static SendTPUEmbeddingGradients créer ( Scope scope, Iterable< Operand < TFloat32 >> entrées, Iterable< Operand < TFloat32 >> learningRates, String config)

Méthode d'usine pour créer une classe encapsulant une nouvelle opération SendTPUEmbeddingGradients.

Paramètres
portée portée actuelle
contributions Une TensorList de dégradés avec lesquels mettre à jour les tables d'intégration. Cet argument a la même longueur et la même forme que la valeur de retour de RecvTPUEmbeddingActivations, mais contient des gradients de perte du modèle par rapport aux activations d'intégration. Les tables d'intégration sont mises à jour à partir de ces dégradés via l'optimiseur spécifié dans la configuration d'intégration TPU donnée à tpu.initialize_system.
Tarifs d'apprentissage Une TensorList de scalaires float32, un pour chaque balise de taux d'apprentissage dynamique : voir les commentaires dans //third_party/tensorflow/core/protobuf/tpu/optimization_parameters.proto. Plusieurs tables peuvent partager la même balise de taux d'apprentissage dynamique, comme spécifié dans la configuration. Si les taux d'apprentissage pour toutes les tables sont constants, cette liste doit être vide.
configuration Protocole TPUEmbeddingConfiguration sérialisé.
Retour
  • une nouvelle instance de SendTPUEmbeddingGradients