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

OutfeedDequeueTupleV2

public final class OutfeedDequeueTupleV2

Retrieve multiple values from the computation outfeed. Device ordinal is a tensor allowing dynamic outfeed.

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

Public Methods

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

Inherited Methods

Public Methods

public static OutfeedDequeueTupleV2 create (Scope scope, Operand<Integer> deviceOrdinal, List<Class<?>> dtypes, List<Shape> shapes)

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

Parameters
scope current scope
deviceOrdinal An int scalar tensor, representing 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.
dtypes The element types of each element in `outputs`.
shapes The shapes of each tensor in `outputs`.
Returns
  • a new instance of OutfeedDequeueTupleV2

public Iterator<Operand<Object>> iterator ()

public List<Output<?>> outputs ()

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