TPUExecuteAndUpdateVariables

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
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.
Iterator<Operand<Object>>
iterator()
List<Output<?>>
results()

Inherited Methods

org.tensorflow.op.PrimitiveOp
final boolean
equals(Object obj)
final int
hashCode()
Operation
op()
Returns the underlying Operation
final String
toString()
boolean
equals(Object arg0)
final Class<?>
getClass()
int
hashCode()
final void
notify()
final void
notifyAll()
String
toString()
final void
wait(long arg0, int arg1)
final void
wait(long arg0)
final void
wait()
void
forEach(Consumer<? super T> arg0)
abstract Iterator<Operand<Object>>
iterator()
Spliterator<Operand<Object>>
spliterator()

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

public Iterator<Operand<Object>> iterator ()

public List<Output<?>> results ()