InfeedEnqueuePrelinearizedBuffer

public final class InfeedEnqueuePrelinearizedBuffer

An op which enqueues prelinearized buffer into TPU infeed.

Nested Classes

class InfeedEnqueuePrelinearizedBuffer.Options Optional attributes for InfeedEnqueuePrelinearizedBuffer  

Constants

String OP_NAME The name of this op, as known by TensorFlow core engine

Public Methods

static InfeedEnqueuePrelinearizedBuffer
create(Scope scope, Operand<?> input, Options... options)
Factory method to create a class wrapping a new InfeedEnqueuePrelinearizedBuffer operation.
static InfeedEnqueuePrelinearizedBuffer.Options
deviceOrdinal(Long deviceOrdinal)

Inherited Methods

Constants

public static final String OP_NAME

The name of this op, as known by TensorFlow core engine

Constant Value: "InfeedEnqueuePrelinearizedBuffer"

Public Methods

public static InfeedEnqueuePrelinearizedBuffer create (Scope scope, Operand<?> input, Options... options)

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

Parameters
scope current scope
input A variant tensor representing linearized output.
options carries optional attributes values
Returns
  • a new instance of InfeedEnqueuePrelinearizedBuffer

public static InfeedEnqueuePrelinearizedBuffer.Options deviceOrdinal (Long deviceOrdinal)

Parameters
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.