org.tensorflow.op.core

Class InfeedEnqueuePrelinearizedBuffer.Options

java.lang.Object

org.tensorflow.op.core.InfeedEnqueuePrelinearizedBuffer.Options

Enclosing class:

InfeedEnqueuePrelinearizedBuffer

public static class InfeedEnqueuePrelinearizedBuffer.Options
extends Object

Optional attributes for InfeedEnqueuePrelinearizedBuffer

Method Summary

Modifier and Type	Method and Description
InfeedEnqueuePrelinearizedBuffer.Options	deviceOrdinal(Long deviceOrdinal)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

deviceOrdinal

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.