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>>
iterator()
List<Output<?>>
results()

Inherited Methods

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 ()