TPUReshardVariables

classe final pública TPUReshardVariables

Opção que reestrutura variáveis de TPU no dispositivo para um estado especificado.

Opção que reestrutura variáveis de TPU no dispositivo para um 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

TPUReshardVariables estáticas

create ( Escopo , Iterable< Operand <?>> vars, Operand <String> newFormatKey, Operand <?> formatStateVar)

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

Métodos herdados

Da classe org.tensorflow.op.PrimitiveOp

booleano final	é igual (objeto obj)
int final	código hash ()
Operação	op () Retorna a `Operation` subjacente
sequência final	para sequenciar ()

Da classe java.lang.Object

boleano	é igual (objeto arg0)
aula final<?>	getClass ()
interno	código hash ()
vazio final	notificar ()
vazio final	notificar todos ()
Corda	para sequenciar ()
vazio final	espere (long arg0, int arg1)
vazio final	espere (arg0 longo)
vazio final	espere ()

Métodos Públicos

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

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

Parâmetros

escopo	escopo atual

Devoluções

uma nova instância de TPUReshardVariables