TPUExecuteAndUpdateVariables

clase final pública TPUExecuteAndUpdateVariables

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

(Opcionalmente) lee 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 a partir 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). Estas salidas del programa son consumidas por estas variables y no aparecerán en la salida de operación. Para el uso interno del compilador de TPU distribuido.

Métodos públicos

TPUExecuteAndUpdateVariables estáticas
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 envuelve 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 envuelve una nueva operación TPUExecuteAndUpdateVariables.

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

public Iterator< Operando <Objeto>> iterador ()

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