Aprenda o que há de mais recente em aprendizado de máquina, IA generativa e muito mais no WiML Symposium 2023 Registre-se

Esta página foi traduzida pela API Cloud Translation.

TPUReshardVariables

classe final pública TPUreshardVariables

Op que refragmenta variáveis de TPU no dispositivo para o estado especificado.

Op que refragmenta variáveis de TPU no dispositivo para o estado especificado. Somente para uso interno.

O estado de fragmentação é representado como a chave da compilação que gerou os programas de fragmentação/desfragmentação junto com o programa principal. new_format_key especifica o estado desejado e format_state_var é o estado atual das variáveis.

Métodos públicos

static TPUreshardVariables

create ( Escopo do escopo, Iterable< Operando <?>> vars, Operando <String> newFormatKey, Operando <?> formatStateVar)

Método de fábrica para criar uma classe que envolve uma nova operação TPUreshardVariables.

Métodos Herdados

Da classe org.tensorflow.op.PrimitiveOp

booleano final	igual a (objeto obj)
final int	código de hash ()
Operação	op () Retorna a `Operation` subjacente
String final	toString ()

Da classe java.lang.Object

boleano	igual a (objeto arg0)
Aula final<?>	getClass ()
int	código de hash ()
vazio final	notifique ()
vazio final	notifiqueTodos ()
Fragmento	toString ()
vazio final	espere (long arg0, int arg1)
vazio final	espere (arg0 longo)
vazio final	espere ()

Métodos públicos

public static TPUReshardVariables create ( Escopo do escopo, Iterable< Operando <?>> vars, Operando <String> newFormatKey, Operando <?> formatStateVar)

Método de fábrica para criar uma classe que envolve uma nova operação TPUreshardVariables.

Parâmetros

alcance	escopo atual

Devoluções

uma nova instância de TPUreshardVariables