TPUExecuteAndUpdateVariables

TPUExecuteAndUpdateVariables public final class

Op que ejecuta un programa con actualizaciones de variables in situ opcionales.

Lee (opcionalmente) las variables del dispositivo, carga y ejecuta un programa de TPU en un dispositivo de TPU y luego (opcionalmente) actualiza las variables en el lugar utilizando las salidas del programa, como se especifica en los atributos device_var_reads_indices (índices de entrada del programa de la lectura directa de variables) y device_var_updates_indices (índices de salida del programa utilizados para actualizar variables, -1 significa sin actualización / solo lectura). Tales salidas de programa son consumidas por estas variables y no aparecerán en la salida de operación. Para uso interno del compilador de TPU distribuido.

Métodos públicos

estáticas TPUExecuteAndUpdateVariables
crear ( Alcance alcance, Iterable < operando <? >> args, operando <String> llave, List <Clase <? >> Tresults, List <Larga> deviceVarReadsIndices, List <Larga> deviceVarUpdatesIndices)
Método de fábrica para crear una clase que envuelva una nueva operación TPUExecuteAndUpdateVariables.
Iterador < operando <Object >>
Lista < salida <? >>

Métodos heredados

Métodos públicos

estáticas pública TPUExecuteAndUpdateVariables crean ( Alcance alcance, Iterable < operando <? >> args, operando <String> llave, List <Clase <? >> Tresults, List <Larga> deviceVarReadsIndices, List <Larga> deviceVarUpdatesIndices)

Método de fábrica para crear una clase que envuelva una nueva operación TPUExecuteAndUpdateVariables.

Parámetros
alcance alcance actual
Devoluciones
  • una nueva instancia de TPUExecuteAndUpdateVariables

pública Iterator < operando <Object >> iterador ()

Lista pública < salida <? >> resultados ()