EnqueueTPUEmbeddingIntegerBatch

public final class EnqueueTPUEmbeddingIntegerBatch

An op that enqueues a list of input batch tensors to TPUEmbedding.

class EnqueueTPUEmbeddingIntegerBatch.Options Optional attributes for EnqueueTPUEmbeddingIntegerBatch

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

static EnqueueTPUEmbeddingIntegerBatch	create(Scope scope, Iterable<Operand<TInt32>> batch, Operand<TString> modeOverride, Options... options) Factory method to create a class wrapping a new EnqueueTPUEmbeddingIntegerBatch operation.
static EnqueueTPUEmbeddingIntegerBatch.Options	deviceOrdinal(Long deviceOrdinal)

From class org.tensorflow.op.RawOp

final boolean	equals(Object obj)
final int	hashCode()
Operation	op() Return this unit of computation as a single `Operation`.
final String	toString()

From class java.lang.Object

From interface org.tensorflow.op.Op

abstract ExecutionEnvironment	env() Return the execution environment this op was created in.
abstract Operation	op() Return this unit of computation as a single `Operation`.

Constants

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

Constant Value: "EnqueueTPUEmbeddingIntegerBatch"

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

scope	current scope
batch	A list of 1D tensors, one for each embedding table, containing the indices into the tables.
modeOverride	A string input that overrides the mode specified in the TPUEmbeddingConfiguration. Supported values are {'unspecified', 'inference', 'training', 'backward_pass_only'}. When set to 'unspecified', the mode set in TPUEmbeddingConfiguration is used, otherwise mode_override is used.
options	carries optional attributes values

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.