Have a question? Connect with the community at the TensorFlow Forum Visit Forum

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

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