org.tensorflow.op.tpu

Class EnqueueTPUEmbeddingIntegerBatch

java.lang.Object

org.tensorflow.op.RawOp

org.tensorflow.op.tpu.EnqueueTPUEmbeddingIntegerBatch

All Implemented Interfaces:

Op

public final class EnqueueTPUEmbeddingIntegerBatch
extends RawOp

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

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	EnqueueTPUEmbeddingIntegerBatch.Inputs
static class	EnqueueTPUEmbeddingIntegerBatch.Options Optional attributes for EnqueueTPUEmbeddingIntegerBatch

Field Summary

Fields

Modifier and Type	Field and Description
static String	OP_NAME The name of this op, as known by TensorFlow core engine

Fields inherited from class org.tensorflow.op.RawOp

operation

Constructor Summary

Constructors

Constructor and Description
EnqueueTPUEmbeddingIntegerBatch(Operation operation)

Method Summary

Modifier and Type	Method and Description
static EnqueueTPUEmbeddingIntegerBatch	create(Scope scope, Iterable<Operand<TInt32>> batch, Operand<TString> modeOverride, EnqueueTPUEmbeddingIntegerBatch.Options... options) Factory method to create a class wrapping a new EnqueueTPUEmbeddingIntegerBatch operation.
static EnqueueTPUEmbeddingIntegerBatch.Options	deviceOrdinal(Long deviceOrdinal) Sets the deviceOrdinal option.

Methods inherited from class org.tensorflow.op.RawOp

equals, hashCode, op, toString

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Methods inherited from interface org.tensorflow.op.Op

env

Field Detail

OP_NAME

public static final String OP_NAME

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

See Also:

Constant Field Values

Constructor Detail

EnqueueTPUEmbeddingIntegerBatch

public EnqueueTPUEmbeddingIntegerBatch(Operation operation)

Method Detail

create

@Endpoint(describeByClass=true)
public static EnqueueTPUEmbeddingIntegerBatch create(Scope scope,
                                                                                     Iterable<Operand<TInt32>> batch,
                                                                                     Operand<TString> modeOverride,
                                                                                     EnqueueTPUEmbeddingIntegerBatch.Options... options)

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

Parameters:

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 attribute values

Returns:

a new instance of EnqueueTPUEmbeddingIntegerBatch

deviceOrdinal

public static EnqueueTPUEmbeddingIntegerBatch.Options deviceOrdinal(Long deviceOrdinal)

Sets the deviceOrdinal option.

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.

Returns:

this Options instance.