TPUExecuteAndUpdateVariables

classe final pública TPUExecuteAndUpdateVariables

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

Ele (opcionalmente) lê as variáveis ​​do dispositivo, carrega e executa um programa TPU em um dispositivo TPU e, em seguida, (opcionalmente) atualiza as variáveis ​​no local usando as saídas do programa, conforme especificado nos atributos device_var_reads_indices (índices de entrada do programa 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 do 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

static TPUExecuteAndUpdateVariables
create ( Escopo do escopo, Iterable< Operando <?>> argumentos, Operando <String> chave, 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 ( Escopo do escopo, Iterable< Operando <?>> argumentos, Operando <String> chave, 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
alcance escopo atual
Devoluções
  • uma nova instância de TPUExecuteAndUpdateVariables

public Iterator< Operando <Object>> iterador ()

public List< Saída <?>> resultados ()