TPUExecuteAndUpdateVariables

clase final pública TPUExecuteAndUpdateVariables

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

TPUE estáticoxecuteAndUpdateVariables
create ( Scope scope, Iterable < Operand <? >> args, Operand <String> key, List <Class <? >> Tresults, List <Long> deviceVarReadsIndices, List <Long> deviceVarUpdatesIndices)
Método de fábrica para crear una clase que envuelva una nueva operación TPUExecuteAndUpdateVariables.
Iterador < Operando <Objeto>>
iterador ()
Lista < Salida <? >>
resultados ()

Métodos heredados

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 crear una clase que envuelva una nueva operación TPUExecuteAndUpdateVariables.

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

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

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