An op that enqueues a list of input batch tensors to TPUEmbedding.
tf.raw_ops.EnqueueTPUEmbeddingIntegerBatch(
batch, mode_override, device_ordinal=-1, name=None
)
Args |
|---|
batch
|
A list of at least 1 Tensor objects with type int32.
A list of 1D tensors, one for each embedding table, containing the
indices into the tables.
|
mode_override
|
A Tensor of type string.
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.
|
device_ordinal
|
An optional int. Defaults to -1.
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.
|
name
|
A name for the operation (optional).
|
Returns |
|---|
|
The created Operation.
|
