TPUReshardVariables

TPUReshardVariables classe final públicas

Op que compartilha novamente as variáveis ​​da TPU no dispositivo para o estado especificado.

Op que compartilha novamente as variáveis ​​da 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 / retirada de fragmentaçã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

estáticos TPUReshardVariables
criar ( Scope 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

Métodos Públicos

public static TPUReshardVariables criar ( Scope 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