Attend the Women in ML Symposium on December 7 Register now

TPUExecute

Stay organized with collections Save and categorize content based on your preferences.
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 ()