TPUReshardVariables

classe finale pubblica TPUReshardVariables

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
create ( Scope scope, Iterable< Operand <?>> vars, Operand <String> newFormatKey, Operand <?> formatStateVar)
Metodo factory per creare una classe che racchiude una nuova operazione TPUReshardVariables.

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