InfeedEnqueuePrelinearizedBuffer.Options

public static class InfeedEnqueuePrelinearizedBuffer.Options

Optional attributes for InfeedEnqueuePrelinearizedBuffer

Public Methods

InfeedEnqueuePrelinearizedBuffer.Options
deviceOrdinal(Long deviceOrdinal)

Inherited Methods

Public Methods

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