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