TPUReshardVariables

clase final pública TPUReshardVariables

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
crear (alcance alcance , Iterable< Operando <?>> vars, Operando <Cadena> newFormatKey, Operando <?> formatStateVar)
Método de fábrica para crear una clase que envuelve una nueva operación TPUReshardVariables.

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