Operazione che riconfigura le variabili TPU sul dispositivo allo stato specificato.
Operazione che riconfigura le variabili TPU sul dispositivo allo stato specificato. Solo per uso interno.
Lo stato di sharding è rappresentato come la chiave della compilazione che ha generato i programmi di sharding/unsharding insieme al programma principale. new_format_key specifica lo stato desiderato e format_state_var è lo stato corrente delle variabili.
Metodi pubblici
Variabili TPUReshard statiche |
Metodi ereditati
Metodi pubblici
creazione di TPUReshardVariables statiche pubbliche ( scope scope, Iterable< Operand <?>> vars, Operand <String> newFormatKey, Operand <?> formatStateVar)
Metodo factory per creare una classe che racchiude una nuova operazione TPUReshardVariables.
Parametri
scopo | ambito attuale |
---|
ritorna
- una nuova istanza di TPUReshardVariables