public static class EnqueueTPUEmbeddingSparseBatch.Inputs extends RawOpInputs<EnqueueTPUEmbeddingSparseBatch>
Modifier and Type | Field and Description |
---|---|
Iterable<Operand<? extends TNumber>> |
aggregationWeights
A list of rank 1 Tensors containing per sample -- i.e.
|
String[] |
combiners
A list of string scalars, one for each embedding table that specify
how to normalize the embedding activations after weighted summation.
|
long |
deviceOrdinal
The TPU device to use.
|
Iterable<Operand<? extends TNumber>> |
embeddingIndices
A list of rank 1 Tensors, indices into the embedding tables.
|
Operand<TString> |
modeOverride
A string input that overrides the mode specified in the
TPUEmbeddingConfiguration.
|
Iterable<Operand<? extends TNumber>> |
sampleIndices
A list of rank 1 Tensors specifying the training example and
feature to which the corresponding embedding_indices and aggregation_weights
values belong.
|
DataType |
T1
The T1 attribute
|
DataType |
T2
The T2 attribute
|
DataType |
T3
The T3 attribute
|
Constructor and Description |
---|
Inputs(GraphOperation op) |
attributeMetadata, attributeNames, attributes, attributeValue, attributeValues, equals, getOutputs, hashCode, toString
public final Iterable<Operand<? extends TNumber>> sampleIndices
public final Iterable<Operand<? extends TNumber>> embeddingIndices
public final Iterable<Operand<? extends TNumber>> aggregationWeights
public final Operand<TString> modeOverride
public final DataType T1
public final DataType T2
public final DataType T3
public final long deviceOrdinal
public final String[] combiners
public Inputs(GraphOperation op)
Copyright © 2015–2022. All rights reserved.