Conozca lo último en aprendizaje automático, IA generativa y más en el Simposio WiML 2023.

Se usó la API de Cloud Translation para traducir esta página.

TPUReshardVariables

clase final pública TPUReshardVariables

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

crear ( Ámbito ámbito, Iterable< Operando <?>> vars, Operando <String> newFormatKey, Operando <?> formatStateVar)

Método de fábrica para crear una clase que envuelve una nueva operación TPUReshardVariables.

Métodos Heredados

De la clase org.tensorflow.op.PrimitiveOp

booleano final	es igual a (Objeto obj)
int final	código hash ()
Operación	op () Devuelve la `Operation` subyacente
cadena final	a la cadena ()

De la clase java.lang.Object

booleano	es igual a (Objeto arg0)
clase final<?>	obtenerClase ()
En t	código hash ()
vacío final	notificar ()
vacío final	notificar a todos ()
Cuerda	a la cadena ()
vacío final	espera (largo arg0, int arg1)
vacío final	espera (largo arg0)
vacío final	espera ()

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