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 |
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