OutfeedDequeueTuple

public final class OutfeedDequeueTuple

Retrieve multiple values from the computation outfeed.

This operation will block indefinitely until data is available. Output `i` corresponds to XLA tuple element `i`.

Nested Classes

class OutfeedDequeueTuple.Options Optional attributes for OutfeedDequeueTuple  

Public Methods

static OutfeedDequeueTuple
create(Scope scope, List<Class<?>> dtypes, List<Shape> shapes, Options... options)
Factory method to create a class wrapping a new OutfeedDequeueTuple operation.
static OutfeedDequeueTuple.Options
deviceOrdinal(Long deviceOrdinal)
Iterator<Operand<Object>>
List<Output<?>>
outputs()
A list of tensors that will be read from the outfeed.

Inherited Methods

Public Methods

public static OutfeedDequeueTuple create (Scope scope, List<Class<?>> dtypes, List<Shape> shapes, Options... options)

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

Parameters
scope current scope
dtypes The element types of each element in `outputs`.
shapes The shapes of each tensor in `outputs`.
options carries optional attributes values
Returns
  • a new instance of OutfeedDequeueTuple

public static OutfeedDequeueTuple.Options deviceOrdinal (Long deviceOrdinal)

Parameters
deviceOrdinal The TPU device to use. This should be -1 when the Op is running on a TPU device, and >= 0 when the Op is running on the CPU device.

public Iterator<Operand<Object>> iterator ()

public List<Output<?>> outputs ()

A list of tensors that will be read from the outfeed.