ExecuteAndUpdateVariables

public final class ExecuteAndUpdateVariables

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.

Constants

String OP_NAME The name of this op, as known by TensorFlow core engine

Public Methods

static ExecuteAndUpdateVariables
create(Scope scope, Iterable<Operand<?>> args, Operand<TString> key, List<Class<? extends TType>> Tresults, List<Long> deviceVarReadsIndices, List<Long> deviceVarUpdatesIndices)
Factory method to create a class wrapping a new ExecuteAndUpdateVariables operation.
Iterator<Operand<TType>>
List<Output<?>>

Inherited Methods

org.tensorflow.op.RawOp
final boolean
equals(Object obj)
final int
Operation
op()
Return this unit of computation as a single Operation.
final String
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()
org.tensorflow.op.Op
abstract ExecutionEnvironment
env()
Return the execution environment this op was created in.
abstract Operation
op()
Return this unit of computation as a single Operation.
abstract void
forEach(Consumer<? super T> arg0)
abstract Iterator<Operand<TType>>
iterator()
abstract Spliterator<Operand<TType>>
spliterator()

Constants

public static final String OP_NAME

The name of this op, as known by TensorFlow core engine

Constant Value: "TPUExecuteAndUpdateVariables"

Public Methods

public static ExecuteAndUpdateVariables create (Scope scope, Iterable<Operand<?>> args, Operand<TString> key, List<Class<? extends TType>> Tresults, List<Long> deviceVarReadsIndices, List<Long> deviceVarUpdatesIndices)

Factory method to create a class wrapping a new ExecuteAndUpdateVariables operation.

Parameters
scope current scope
Returns
  • a new instance of ExecuteAndUpdateVariables

public Iterator<Operand<TType>> iterator ()

public List<Output<?>> results ()