Op que vuelve a fragmentar las variables de TPU en el dispositivo al estado especificado.
Op 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
Variables estáticas de TPUReshard |
Métodos Heredados
Métodos públicos
public static TPUReshardVariables create ( Scope scope, Iterable< Operand <?>> vars, Operand <String> newFormatKey, Operand <?> 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