TPUReshardVariables

classe finale publique TPUReshardVariables

Opération qui repartage les variables TPU sur l'appareil à l'état spécifié.

Opération qui repartage les variables TPU sur l'appareil à l'état spécifié. Usage interne uniquement.

L'état de partitionnement est représenté comme la clé de la compilation qui a généré les programmes de partitionnement/défragmentation avec le programme principal. new_format_key spécifie l'état souhaité et format_state_var est l'état actuel des variables.

Méthodes publiques

Variables TPUReshard statiques
créer ( Scope scope, Iterable< Operand <?>> vars, Operand <String> newFormatKey, Operand <?> formatStateVar)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération TPUReshardVariables.

Méthodes héritées

Méthodes publiques

public static TPUReshardVariables créer ( Scope scope, Iterable< Operand <?>> vars, Operand <String> newFormatKey, Operand <?> formatStateVar)

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

Paramètres
portée portée actuelle
Retour
  • une nouvelle instance de TPUReshardVariables