Opción que vuelve a fragmentar las variables de TPU en el dispositivo al estado especificado.
Opción que vuelve a fragmentar las variables de TPU en el dispositivo al estado especificado. Solo para uso interno.
El estado de fragmentación se representa como la clave de la compilación que generó los programas de fragmentación/desfragmentación junto con el programa principal. new_format_key especifica el estado deseado y format_state_var es el estado actual de las variables.
Métodos públicos
TPU estáticaReshardVariables |
Métodos heredados
Métodos públicos
creación pública estática de TPUReshardVariables (alcance de alcance , Iterable < Operando <? >> vars, Operando <String> newFormatKey, Operando <?> formatStateVar)
Método de fábrica para crear una clase que envuelve una nueva operación TPUReshardVariables.
Parámetros
alcance | alcance actual |
---|
Devoluciones
- una nueva instancia de TPUReshardVariables