public final class
TPUExecuteAndUpdateVariables
Op that executes a program with optional in-place variable updates.
It (optionally) reads device variables, loads and executes a TPU program on a TPU device, and then (optionally) in-place updates variables using the program outputs, as specified in attributes device_var_reads_indices (program input indices from directly reading variables) and device_var_updates_indices (program output indices used to update variables, -1 means no-update/read-only). Such program outputs are consumed by these variables will not appear in the op output. For the internal use of the distributed TPU compiler.
Public Methods
| static TPUExecuteAndUpdateVariables | |
| Iterator<Operand<Object>> |
iterator()
|
| List<Output<?>> |
results()
|
Inherited Methods
Public Methods
public static TPUExecuteAndUpdateVariables create (Scope scope, Iterable<Operand<?>> args, Operand<String> key, List<Class<?>> Tresults, List<Long> deviceVarReadsIndices, List<Long> deviceVarUpdatesIndices)
Factory method to create a class wrapping a new TPUExecuteAndUpdateVariables operation.
Parameters
| scope | current scope |
|---|
Returns
- a new instance of TPUExecuteAndUpdateVariables