org.tensorflow.op.core

Class EnqueueTPUEmbeddingIntegerBatch.Options

java.lang.Object

org.tensorflow.op.core.EnqueueTPUEmbeddingIntegerBatch.Options

Enclosing class:

EnqueueTPUEmbeddingIntegerBatch

public static class EnqueueTPUEmbeddingIntegerBatch.Options
extends Object

Optional attributes for EnqueueTPUEmbeddingIntegerBatch

Method Summary

Modifier and Type	Method and Description
EnqueueTPUEmbeddingIntegerBatch.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 EnqueueTPUEmbeddingIntegerBatch.Options deviceOrdinal(Long deviceOrdinal)

Parameters:

deviceOrdinal - The TPU device to use. Should be >= 0 and less than the number of TPU cores in the task on which the node is placed.