public final class InfeedEnqueue extends PrimitiveOp
Modifier and Type | Class and Description |
---|---|
static class |
InfeedEnqueue.Options
Optional attributes for
InfeedEnqueue |
operation
Modifier and Type | Method and Description |
---|---|
static <T> InfeedEnqueue |
create(Scope scope,
Operand<T> input,
InfeedEnqueue.Options... options)
Factory method to create a class wrapping a new InfeedEnqueue operation.
|
static InfeedEnqueue.Options |
deviceOrdinal(Long deviceOrdinal) |
static InfeedEnqueue.Options |
layout(List<Long> layout) |
static InfeedEnqueue.Options |
shape(Shape shape) |
equals, hashCode, op, toString
public static <T> InfeedEnqueue create(Scope scope, Operand<T> input, InfeedEnqueue.Options... options)
scope
- current scopeinput
- A tensor that will be provided using the infeed mechanism.options
- carries optional attributes valuespublic static InfeedEnqueue.Options shape(Shape shape)
shape
- The shape of the tensor.public static InfeedEnqueue.Options layout(List<Long> layout)
layout
- A vector holding the requested layout in minor-to-major sequence.
If a layout attribute is passed, but its values are all -1, the layout will
be computed by the infeed operation.public static InfeedEnqueue.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.