Advertencia: esta API está obsoleta y se eliminará en una versión futura de TensorFlow una vez que el reemplazo sea ​​estable.

TPUReshardVariables

Organiza tus páginas con colecciones Guarda y categoriza el contenido según tus preferencias.
TPUReshardVariables public final class

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
crear ( 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.

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