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`.

Constants

String OP_NAME The name of this op, as known by TensorFlow core engine

Public Methods

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

Inherited Methods

Constants

public static final String OP_NAME

The name of this op, as known by TensorFlow core engine

Constant Value: "OutfeedDequeueTupleV2"

Public Methods

public static OutfeedDequeueTupleV2 create (Scope scope, Operand<TInt32> deviceOrdinal, List<Class<? extends TType>> 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<TType>> iterator ()

public List<Output<?>> outputs ()

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