Warning: This API is deprecated and will be removed in a future
version of TensorFlow after
the replacement is stable.
EnqueueTPUEmbeddingBatch
Stay organized with collections
Save and categorize content based on your preferences.
An op that enqueues a list of input batch tensors to TPUEmbedding.
An op that enqueues a list of input batch tensors to TPUEmbedding.
Inherited Methods
From class
java.lang.Object
|
boolean
|
equals(Object arg0)
|
|
final
Class<?>
|
getClass()
|
|
int
|
hashCode()
|
|
final
void
|
notify()
|
|
final
void
|
notifyAll()
|
|
String
|
toString()
|
|
final
void
|
wait(long arg0, int arg1)
|
|
final
void
|
wait(long arg0)
|
|
final
void
|
wait()
|
Public Methods
Parameters
| combiners |
A list of string scalars, one for each embedding table that specify
how to normalize the embedding activations after weighted summation.
Supported combiners are 'mean', 'sum', or 'sqrtn'. It is invalid to have
the sum of the weights be 0 for 'mean' or the sum of the squared weights be
0 for 'sqrtn'. If combiners isn't passed, the default is to use 'sum' for
all tables.
|
Factory method to create a class wrapping a new EnqueueTPUEmbeddingBatch operation.
Parameters
| scope |
current scope |
| batch |
A list of 1D tensors, one for each embedding table, containing the
batch inputs encoded as dist_belief.SparseFeatures protos. If the weight
field in the SparseFeatures proto is not populated for an ID, a weight of
1.0 is assumed. |
| 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 |
Returns
- a new instance of EnqueueTPUEmbeddingBatch
Parameters
| deviceOrdinal |
The TPU device to use. This should be -1 when the Op
is running on a TPU device, and >= 0 when the Op is running on the CPU
device.
|
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.
Last updated 2022-05-17 UTC.
[[["Easy to understand","easyToUnderstand","thumb-up"],["Solved my problem","solvedMyProblem","thumb-up"],["Other","otherUp","thumb-up"]],[["Missing the information I need","missingTheInformationINeed","thumb-down"],["Too complicated / too many steps","tooComplicatedTooManySteps","thumb-down"],["Out of date","outOfDate","thumb-down"],["Samples / code issue","samplesCodeIssue","thumb-down"],["Other","otherDown","thumb-down"]],["Last updated 2022-05-17 UTC."],[],[],null,["# EnqueueTPUEmbeddingBatch\n\npublic final class **EnqueueTPUEmbeddingBatch** \nAn op that enqueues a list of input batch tensors to TPUEmbedding.\n\n\nAn op that enqueues a list of input batch tensors to TPUEmbedding.\n\n\u003cbr /\u003e\n\n### Nested Classes\n\n|-------|---|---|--------------------------------------------------------------------------------------------------------------------|\n| class | [EnqueueTPUEmbeddingBatch.Options](/api_docs/java/org/tensorflow/op/core/EnqueueTPUEmbeddingBatch.Options) || Optional attributes for [EnqueueTPUEmbeddingBatch](/api_docs/java/org/tensorflow/op/core/EnqueueTPUEmbeddingBatch) |\n\n### Public Methods\n\n|-------------------------------------------------------------------------------------------------------------------|------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|\n| static [EnqueueTPUEmbeddingBatch.Options](/api_docs/java/org/tensorflow/op/core/EnqueueTPUEmbeddingBatch.Options) | [combiners](/api_docs/java/org/tensorflow/op/core/EnqueueTPUEmbeddingBatch#combiners(java.util.List\u003cjava.lang.String\u003e))(List\\\u003cString\\\u003e combiners) |\n| static [EnqueueTPUEmbeddingBatch](/api_docs/java/org/tensorflow/op/core/EnqueueTPUEmbeddingBatch) | [create](/api_docs/java/org/tensorflow/op/core/EnqueueTPUEmbeddingBatch#create(org.tensorflow.op.Scope,%20java.lang.Iterable\u003corg.tensorflow.Operand\u003cjava.lang.String\u003e\u003e,%20org.tensorflow.Operand\u003cjava.lang.String\u003e,%20org.tensorflow.op.core.EnqueueTPUEmbeddingBatch.Options...))([Scope](/api_docs/java/org/tensorflow/op/Scope) scope, Iterable\\\u003c[Operand](/api_docs/java/org/tensorflow/Operand)\\\u003cString\\\u003e\\\u003e batch, [Operand](/api_docs/java/org/tensorflow/Operand)\\\u003cString\\\u003e modeOverride, [Options...](/api_docs/java/org/tensorflow/op/core/EnqueueTPUEmbeddingBatch.Options) options) Factory method to create a class wrapping a new EnqueueTPUEmbeddingBatch operation. |\n| static [EnqueueTPUEmbeddingBatch.Options](/api_docs/java/org/tensorflow/op/core/EnqueueTPUEmbeddingBatch.Options) | [deviceOrdinal](/api_docs/java/org/tensorflow/op/core/EnqueueTPUEmbeddingBatch#deviceOrdinal(java.lang.Long))(Long deviceOrdinal) |\n\n### Inherited Methods\n\nFrom class [org.tensorflow.op.PrimitiveOp](/api_docs/java/org/tensorflow/op/PrimitiveOp) \n\n|------------------------------------------------------|---------------------------------------------------------------------------------------------------------------------------------------|\n| final boolean | [equals](/api_docs/java/org/tensorflow/op/PrimitiveOp#equals(java.lang.Object))(Object obj) |\n| final int | [hashCode](/api_docs/java/org/tensorflow/op/PrimitiveOp#hashCode())() |\n| [Operation](/api_docs/java/org/tensorflow/Operation) | [op](/api_docs/java/org/tensorflow/op/PrimitiveOp#op())() Returns the underlying [Operation](/api_docs/java/org/tensorflow/Operation) |\n| final String | [toString](/api_docs/java/org/tensorflow/op/PrimitiveOp#toString())() |\n\nFrom class java.lang.Object \n\n|------------------|---------------------------|\n| boolean | equals(Object arg0) |\n| final Class\\\u003c?\\\u003e | getClass() |\n| int | hashCode() |\n| final void | notify() |\n| final void | notifyAll() |\n| String | toString() |\n| final void | wait(long arg0, int arg1) |\n| final void | wait(long arg0) |\n| final void | wait() |\n\nPublic Methods\n--------------\n\n#### public static [EnqueueTPUEmbeddingBatch.Options](/api_docs/java/org/tensorflow/op/core/EnqueueTPUEmbeddingBatch.Options)\n**combiners**\n(List\\\u003cString\\\u003e combiners)\n\n\u003cbr /\u003e\n\n##### Parameters\n\n| combiners | A list of string scalars, one for each embedding table that specify how to normalize the embedding activations after weighted summation. Supported combiners are 'mean', 'sum', or 'sqrtn'. It is invalid to have the sum of the weights be 0 for 'mean' or the sum of the squared weights be 0 for 'sqrtn'. If combiners isn't passed, the default is to use 'sum' for all tables. |\n|-----------|-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|\n\n#### public static [EnqueueTPUEmbeddingBatch](/api_docs/java/org/tensorflow/op/core/EnqueueTPUEmbeddingBatch)\n**create**\n([Scope](/api_docs/java/org/tensorflow/op/Scope) scope, Iterable\\\u003c[Operand](/api_docs/java/org/tensorflow/Operand)\\\u003cString\\\u003e\\\u003e batch, [Operand](/api_docs/java/org/tensorflow/Operand)\\\u003cString\\\u003e modeOverride, [Options...](/api_docs/java/org/tensorflow/op/core/EnqueueTPUEmbeddingBatch.Options) options)\n\nFactory method to create a class wrapping a new EnqueueTPUEmbeddingBatch operation. \n\n##### Parameters\n\n| scope | current scope |\n| batch | A list of 1D tensors, one for each embedding table, containing the batch inputs encoded as dist_belief.SparseFeatures protos. If the weight field in the SparseFeatures proto is not populated for an ID, a weight of 1.0 is assumed. |\n| 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. |\n| options | carries optional attributes values |\n|--------------|----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|\n\n##### Returns\n\n- a new instance of EnqueueTPUEmbeddingBatch \n\n#### public static [EnqueueTPUEmbeddingBatch.Options](/api_docs/java/org/tensorflow/op/core/EnqueueTPUEmbeddingBatch.Options)\n**deviceOrdinal**\n(Long deviceOrdinal)\n\n\u003cbr /\u003e\n\n##### Parameters\n\n| deviceOrdinal | The TPU device to use. This should be -1 when the Op is running on a TPU device, and \\\u003e= 0 when the Op is running on the CPU device. |\n|---------------|--------------------------------------------------------------------------------------------------------------------------------------|"]]
