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 |
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