TPUExecuteAndUpdateVariables

TPUExecuteAndUpdateVariables classe final públicas

Op que executa um programa com atualizações opcionais de variáveis ​​no local.

Ele (opcionalmente) lê variáveis ​​de dispositivo, carrega e executa um programa TPU em um dispositivo TPU e, em seguida, (opcionalmente) atualiza variáveis ​​no local usando as saídas do programa, conforme especificado nos atributos device_var_reads_indices (índices de entrada do programa a partir de variáveis ​​de leitura direta) e device_var_updates_indices (índices de saída do programa usados ​​para atualizar variáveis, -1 significa sem atualização / somente leitura). Essas saídas de programa são consumidas por essas variáveis ​​e não aparecerão na saída op. Para uso interno do compilador TPU distribuído.

Métodos Públicos

estáticos TPUExecuteAndUpdateVariables
criar ( Scope escopo, Iterable < Operando <? >> args, Operando <String> chave, List <Class <? >> Tresults, List <longo> deviceVarReadsIndices, List <longo> deviceVarUpdatesIndices)
Método de fábrica para criar uma classe que envolve uma nova operação TPUExecuteAndUpdateVariables.
Iterator < Operando <Object >>
List < Output <? >>

Métodos herdados

Métodos Públicos

public static TPUExecuteAndUpdateVariables criar ( Scope escopo, Iterable < Operando <? >> args, Operando <String> chave, List <Class <? >> Tresults, List <longo> deviceVarReadsIndices, List <longo> deviceVarUpdatesIndices)

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

Parâmetros
alcance escopo atual
Devoluções
  • uma nova instância de TPUExecuteAndUpdateVariables

pública Iterator < Operando <Object >> iterator ()

Lista pública < saída <? >> resultados ()