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.
Constants
| String |
OP_NAME |
The name of this op, as known by TensorFlow core engine |
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()
|
Constants
public
static
final
String
OP_NAME
The name of this op, as known by TensorFlow core engine
Constant Value:
"EnqueueTPUEmbeddingIntegerBatch"
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 2021-11-29 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 2021-11-29 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](/jvm/api_docs/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingIntegerBatch.Options) || Optional attributes for [EnqueueTPUEmbeddingIntegerBatch](/jvm/api_docs/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingIntegerBatch) |\n\n### Constants\n\n|--------|---------------------------------------------------------------------------------------------|---------------------------------------------------------|\n| String | [OP_NAME](/jvm/api_docs/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingIntegerBatch#OP_NAME) | The name of this op, as known by TensorFlow core engine |\n\n### Public Methods\n\n|------------------------------------------------------------------------------------------------------------------------------------|--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|\n| static [EnqueueTPUEmbeddingIntegerBatch](/jvm/api_docs/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingIntegerBatch) | [create](/jvm/api_docs/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingIntegerBatch#create(org.tensorflow.op.Scope, java.lang.Iterable\u003corg.tensorflow.Operand\u003corg.tensorflow.types.TInt32\u003e\u003e, org.tensorflow.Operand\u003corg.tensorflow.types.TString\u003e, org.tensorflow.op.tpu.EnqueueTPUEmbeddingIntegerBatch.Options...))([Scope](/jvm/api_docs/java/org/tensorflow/op/Scope) scope, Iterable\\\u003c[Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003c[TInt32](/jvm/api_docs/java/org/tensorflow/types/TInt32)\\\u003e\\\u003e batch, [Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003c[TString](/jvm/api_docs/java/org/tensorflow/types/TString)\\\u003e modeOverride, [Options...](/jvm/api_docs/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingIntegerBatch.Options) options) Factory method to create a class wrapping a new EnqueueTPUEmbeddingIntegerBatch operation. |\n| static [EnqueueTPUEmbeddingIntegerBatch.Options](/jvm/api_docs/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingIntegerBatch.Options) | [deviceOrdinal](/jvm/api_docs/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingIntegerBatch#deviceOrdinal(java.lang.Long))(Long deviceOrdinal) |\n\n### Inherited Methods\n\nFrom class [org.tensorflow.op.RawOp](/jvm/api_docs/java/org/tensorflow/op/RawOp) \n\n|----------------------------------------------------------|---------------------------------------------------------------------------------------------------------------------------------------------------------------|\n| final boolean | [equals](/jvm/api_docs/java/org/tensorflow/op/RawOp#equals(java.lang.Object))(Object obj) |\n| final int | [hashCode](/jvm/api_docs/java/org/tensorflow/op/RawOp#hashCode())() |\n| [Operation](/jvm/api_docs/java/org/tensorflow/Operation) | [op](/jvm/api_docs/java/org/tensorflow/op/RawOp#op())() Return this unit of computation as a single [Operation](/jvm/api_docs/java/org/tensorflow/Operation). |\n| final String | [toString](/jvm/api_docs/java/org/tensorflow/op/RawOp#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\nFrom interface [org.tensorflow.op.Op](/jvm/api_docs/java/org/tensorflow/op/Op) \n\n|-----------------------------------------------------------------------------------------|------------------------------------------------------------------------------------------------------------------------------------------------------------|\n| abstract [ExecutionEnvironment](/jvm/api_docs/java/org/tensorflow/ExecutionEnvironment) | [env](/jvm/api_docs/java/org/tensorflow/op/Op#env())() Return the execution environment this op was created in. |\n| abstract [Operation](/jvm/api_docs/java/org/tensorflow/Operation) | [op](/jvm/api_docs/java/org/tensorflow/op/Op#op())() Return this unit of computation as a single [Operation](/jvm/api_docs/java/org/tensorflow/Operation). |\n\nConstants\n---------\n\n#### public static final String\n**OP_NAME**\n\nThe name of this op, as known by TensorFlow core engine \nConstant Value: \"EnqueueTPUEmbeddingIntegerBatch\"\n\nPublic Methods\n--------------\n\n#### public static [EnqueueTPUEmbeddingIntegerBatch](/jvm/api_docs/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingIntegerBatch)\n**create**\n([Scope](/jvm/api_docs/java/org/tensorflow/op/Scope) scope, Iterable\\\u003c[Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003c[TInt32](/jvm/api_docs/java/org/tensorflow/types/TInt32)\\\u003e\\\u003e batch, [Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003c[TString](/jvm/api_docs/java/org/tensorflow/types/TString)\\\u003e modeOverride, [Options...](/jvm/api_docs/java/org/tensorflow/op/tpu/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](/jvm/api_docs/java/org/tensorflow/op/tpu/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|---------------|-----------------------------------------------------------------------------------------------------------------------|"]]
