Op que vuelve a compartir las variables de TPU en el dispositivo al estado especificado.
Op que vuelve a compartir 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 / eliminación de fragmentació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
estáticas TPUReshardVariables |
Métodos heredados
Métodos públicos
estáticas pública TPUReshardVariables crean ( Alcance alcance, Iterable < operando <? >> Vars, operando <String> newFormatKey, operando <?> formatStateVar)
Método de fábrica para crear una clase que envuelva una nueva operación TPUReshardVariables.
Parámetros
alcance | alcance actual |
---|
Devoluciones
- una nueva instancia de TPUReshardVariables