public final class InfeedEnqueueTuple extends PrimitiveOp
Modifier and Type | Class and Description |
---|---|
static class |
InfeedEnqueueTuple.Options
Optional attributes for
InfeedEnqueueTuple |
operation
Modifier and Type | Method and Description |
---|---|
static InfeedEnqueueTuple |
create(Scope scope,
Iterable<Operand<?>> inputs,
List<Shape> shapes,
InfeedEnqueueTuple.Options... options)
Factory method to create a class wrapping a new InfeedEnqueueTuple operation.
|
static InfeedEnqueueTuple.Options |
deviceOrdinal(Long deviceOrdinal) |
static InfeedEnqueueTuple.Options |
layouts(List<Long> layouts) |
equals, hashCode, op, toString
public static InfeedEnqueueTuple create(Scope scope, Iterable<Operand<?>> inputs, List<Shape> shapes, InfeedEnqueueTuple.Options... options)
scope
- current scopeinputs
- A list of tensors that will be provided using the infeed mechanism.shapes
- The shapes of each tensor in `inputs`.options
- carries optional attributes valuespublic static InfeedEnqueueTuple.Options layouts(List<Long> layouts)
layouts
- A vector holding the requested layout in minor-to-major sequence for
all the tuple shapes, in the order the shapes appear in the "shapes" input.
The layout elements for a sub-shape can be set to -1, in which case the
corresponding layout will be computed by the infeed operation.public static InfeedEnqueueTuple.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.