ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

TPUExecute

public final class TPUExecute

Op that loads and executes a TPU program on a TPU device.

For the internal use of the distributed TPU compiler.

Public Methods

static TPUExecute
create(Scope scope, Iterable<Operand<?>> args, Operand<String> key, List<Class<?>> Tresults)
Factory method to create a class wrapping a new TPUExecute operation.
Iterator<Operand<Object>>
List<Output<?>>

Inherited Methods

Public Methods

public static TPUExecute create (Scope scope, Iterable<Operand<?>> args, Operand<String> key, List<Class<?>> Tresults)

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

Parameters
scope current scope
Returns
  • a new instance of TPUExecute

public Iterator<Operand<Object>> iterator ()

public List<Output<?>> results ()