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 |
Métodos herdados
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