Warning: This API is deprecated and will be removed in a future
version of TensorFlow after
the replacement is stable.
EnqueueTPUEmbeddingIntegerBatch
Stay organized with collections
Save and categorize content based on your preferences.
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
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 attributes values |
Returns
- a new instance of EnqueueTPUEmbeddingIntegerBatch
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.
|
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-02-12 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-02-12 UTC."],[],[],null,["# EnqueueTPUEmbeddingIntegerBatch\n\npublic final class **EnqueueTPUEmbeddingIntegerBatch** \nAn op that enqueues a list of input batch tensors to TPUEmbedding. \n\n### Nested Classes\n\n|-------|---|---|----------------------------------------------------------------------------------------------------------------------------------|\n| class | [EnqueueTPUEmbeddingIntegerBatch.Options](/api_docs/java/org/tensorflow/op/core/EnqueueTPUEmbeddingIntegerBatch.Options) || Optional attributes for [EnqueueTPUEmbeddingIntegerBatch](/api_docs/java/org/tensorflow/op/core/EnqueueTPUEmbeddingIntegerBatch) |\n\n### Public Methods\n\n|---------------------------------------------------------------------------------------------------------------------------------|------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|\n| static [EnqueueTPUEmbeddingIntegerBatch](/api_docs/java/org/tensorflow/op/core/EnqueueTPUEmbeddingIntegerBatch) | [create](/api_docs/java/org/tensorflow/op/core/EnqueueTPUEmbeddingIntegerBatch#create(org.tensorflow.op.Scope,%20java.lang.Iterable\u003corg.tensorflow.Operand\u003cjava.lang.Integer\u003e\u003e,%20org.tensorflow.Operand\u003cjava.lang.String\u003e,%20org.tensorflow.op.core.EnqueueTPUEmbeddingIntegerBatch.Options...))([Scope](/api_docs/java/org/tensorflow/op/Scope) scope, Iterable\\\u003c[Operand](/api_docs/java/org/tensorflow/Operand)\\\u003cInteger\\\u003e\\\u003e batch, [Operand](/api_docs/java/org/tensorflow/Operand)\\\u003cString\\\u003e modeOverride, [Options...](/api_docs/java/org/tensorflow/op/core/EnqueueTPUEmbeddingIntegerBatch.Options) options) Factory method to create a class wrapping a new EnqueueTPUEmbeddingIntegerBatch operation. |\n| static [EnqueueTPUEmbeddingIntegerBatch.Options](/api_docs/java/org/tensorflow/op/core/EnqueueTPUEmbeddingIntegerBatch.Options) | [deviceOrdinal](/api_docs/java/org/tensorflow/op/core/EnqueueTPUEmbeddingIntegerBatch#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 [EnqueueTPUEmbeddingIntegerBatch](/api_docs/java/org/tensorflow/op/core/EnqueueTPUEmbeddingIntegerBatch)\n**create**\n([Scope](/api_docs/java/org/tensorflow/op/Scope) scope, Iterable\\\u003c[Operand](/api_docs/java/org/tensorflow/Operand)\\\u003cInteger\\\u003e\\\u003e batch, [Operand](/api_docs/java/org/tensorflow/Operand)\\\u003cString\\\u003e modeOverride, [Options...](/api_docs/java/org/tensorflow/op/core/EnqueueTPUEmbeddingIntegerBatch.Options) options)\n\nFactory method to create a class wrapping a new EnqueueTPUEmbeddingIntegerBatch operation. \n\n##### Parameters\n\n| scope | current scope |\n| batch | A list of 1D tensors, one for each embedding table, containing the indices into the tables. |\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 EnqueueTPUEmbeddingIntegerBatch \n\n#### public static [EnqueueTPUEmbeddingIntegerBatch.Options](/api_docs/java/org/tensorflow/op/core/EnqueueTPUEmbeddingIntegerBatch.Options)\n**deviceOrdinal**\n(Long deviceOrdinal)\n\n\u003cbr /\u003e\n\n##### Parameters\n\n| deviceOrdinal | The TPU device to use. Should be \\\u003e= 0 and less than the number of TPU cores in the task on which the node is placed. |\n|---------------|-----------------------------------------------------------------------------------------------------------------------|"]]
