Attend the Women in ML Symposium on December 7 Register now

InfeedDequeueTuple

public final class InfeedDequeueTuple

Fetches multiple values from infeed as an XLA tuple.

Public Methods

static InfeedDequeueTuple
create(Scope scope, List<Class<?>> dtypes, List<Shape> shapes)
Factory method to create a class wrapping a new InfeedDequeueTuple operation.
Iterator<Operand<Object>>
List<Output<?>>
outputs()
A list of tensors that will be provided using the infeed mechanism.

Inherited Methods

Public Methods

public static InfeedDequeueTuple create (Scope scope, List<Class<?>> dtypes, List<Shape> shapes)

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

Parameters
scope current scope
dtypes The element types of each element in `outputs`.
shapes The shapes of each tensor in `outputs`.
Returns
  • a new instance of InfeedDequeueTuple

public Iterator<Operand<Object>> iterator ()

public List<Output<?>> outputs ()

A list of tensors that will be provided using the infeed mechanism.