public final class OutfeedDequeueTuple extends PrimitiveOp implements Iterable<Operand<Object>>
This operation will block indefinitely until data is available. Output `i` corresponds to XLA tuple element `i`.
Modifier and Type | Class and Description |
---|---|
static class |
OutfeedDequeueTuple.Options
Optional attributes for
OutfeedDequeueTuple |
operation
Modifier and Type | Method and Description |
---|---|
static OutfeedDequeueTuple |
create(Scope scope,
List<Class<?>> dtypes,
List<Shape> shapes,
OutfeedDequeueTuple.Options... options)
Factory method to create a class wrapping a new OutfeedDequeueTuple operation.
|
static OutfeedDequeueTuple.Options |
deviceOrdinal(Long deviceOrdinal) |
Iterator<Operand<Object>> |
iterator() |
List<Output<?>> |
outputs()
A list of tensors that will be read from the outfeed.
|
equals, hashCode, op, toString
clone, finalize, getClass, notify, notifyAll, wait, wait, wait
forEach, spliterator
public static OutfeedDequeueTuple create(Scope scope, List<Class<?>> dtypes, List<Shape> shapes, OutfeedDequeueTuple.Options... options)
scope
- current scopedtypes
- The element types of each element in `outputs`.shapes
- The shapes of each tensor in `outputs`.options
- carries optional attributes valuespublic static OutfeedDequeueTuple.Options deviceOrdinal(Long deviceOrdinal)
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.Copyright © 2022. All rights reserved.