public static class EnqueueTPUEmbeddingIntegerBatch.Inputs extends RawOpInputs<EnqueueTPUEmbeddingIntegerBatch>
Modifier and Type | Field and Description |
---|---|
Iterable<Operand<TInt32>> |
batch
A list of 1D tensors, one for each embedding table, containing the
indices into the tables.
|
long |
deviceOrdinal
The TPU device to use.
|
Operand<TString> |
modeOverride
A string input that overrides the mode specified in the
TPUEmbeddingConfiguration.
|
Constructor and Description |
---|
Inputs(GraphOperation op) |
attributeMetadata, attributeNames, attributes, attributeValue, attributeValues, equals, getOutputs, hashCode, toString
public final Iterable<Operand<TInt32>> batch
public final Operand<TString> modeOverride
public final long deviceOrdinal
public Inputs(GraphOperation op)
Copyright © 2015–2022. All rights reserved.