public final class InfeedEnqueuePrelinearizedBuffer extends PrimitiveOp
Modifier and Type | Class and Description |
---|---|
static class |
InfeedEnqueuePrelinearizedBuffer.Options
Optional attributes for
InfeedEnqueuePrelinearizedBuffer |
operation
Modifier and Type | Method and Description |
---|---|
static InfeedEnqueuePrelinearizedBuffer |
create(Scope scope,
Operand<?> input,
InfeedEnqueuePrelinearizedBuffer.Options... options)
Factory method to create a class wrapping a new InfeedEnqueuePrelinearizedBuffer operation.
|
static InfeedEnqueuePrelinearizedBuffer.Options |
deviceOrdinal(Long deviceOrdinal) |
equals, hashCode, op, toString
public static InfeedEnqueuePrelinearizedBuffer create(Scope scope, Operand<?> input, InfeedEnqueuePrelinearizedBuffer.Options... options)
scope
- current scopeinput
- A variant tensor representing linearized output.options
- carries optional attributes valuespublic static InfeedEnqueuePrelinearizedBuffer.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.