TPUExecuteAndUpdateVariables

classe final pública TPUExecuteAndUpdateVariables

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 da leitura direta de variáveis) 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 do programa são consumidas por essas variáveis ​​e não aparecerão na saída operacional. Para uso interno do compilador TPU distribuído.

Métodos Públicos

TPUExecuteAndUpdateVariables estáticos
create ( Escopo do escopo , Iterable< Operand <?>> args, Operand <String> key, List<Class<?>> Tresults, List<Long> deviceVarReadsIndices, List<Long> deviceVarUpdatesIndices)
Método de fábrica para criar uma classe que envolve uma nova operação TPUExecuteAndUpdateVariables.
Iterador< Operando <Objeto>>
Lista< Saída <?>>

Métodos herdados

Métodos Públicos

public static TPUExecuteAndUpdateVariables create ( Scope scope, Iterable< Operand <?>> args, Operand <String> key, List<Class<?>> Tresults, List<Long> deviceVarReadsIndices, List<Long> deviceVarUpdatesIndices)

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

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

iterador público< Operando <Objeto>> iterador ()

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