SendTPUEmbeddingGradients
Stay organized with collections
Save and categorize content based on your preferences.
Performs gradient updates of embedding tables.
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:
"SendTPUEmbeddingGradients"
Public Methods
Factory method to create a class wrapping a new SendTPUEmbeddingGradients operation.
Parameters
| scope |
current scope |
| inputs |
A TensorList of gradients with which to update embedding tables.
This argument has the same length and shapes as the return value of
RecvTPUEmbeddingActivations, but contains gradients of the model's loss
with respect to the embedding activations. The embedding tables are updated
from these gradients via the optimizer specified in the TPU embedding
configuration given to tpu.initialize_system. |
| learningRates |
A TensorList of float32 scalars, one for each dynamic learning
rate tag: see the comments in
//third_party/tensorflow/core/protobuf/tpu/optimization_parameters.proto.
Multiple tables can share the same dynamic learning rate tag as specified
in the configuration. If the learning rates for all tables are constant,
this list should be empty. |
| config |
Serialized TPUEmbeddingConfiguration proto. |
Returns
- a new instance of SendTPUEmbeddingGradients
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,["# SendTPUEmbeddingGradients\n\npublic final class **SendTPUEmbeddingGradients** \nPerforms gradient updates of embedding tables. \n\n### Constants\n\n|--------|---------------------------------------------------------------------------------------|---------------------------------------------------------|\n| String | [OP_NAME](/jvm/api_docs/java/org/tensorflow/op/tpu/SendTPUEmbeddingGradients#OP_NAME) | The name of this op, as known by TensorFlow core engine |\n\n### Public Methods\n\n|--------------------------------------------------------------------------------------------------------|----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|\n| static [SendTPUEmbeddingGradients](/jvm/api_docs/java/org/tensorflow/op/tpu/SendTPUEmbeddingGradients) | [create](/jvm/api_docs/java/org/tensorflow/op/tpu/SendTPUEmbeddingGradients#create(org.tensorflow.op.Scope, java.lang.Iterable\u003corg.tensorflow.Operand\u003corg.tensorflow.types.TFloat32\u003e\u003e, java.lang.Iterable\u003corg.tensorflow.Operand\u003corg.tensorflow.types.TFloat32\u003e\u003e, java.lang.String))([Scope](/jvm/api_docs/java/org/tensorflow/op/Scope) scope, Iterable\\\u003c[Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003c[TFloat32](/jvm/api_docs/java/org/tensorflow/types/TFloat32)\\\u003e\\\u003e inputs, Iterable\\\u003c[Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003c[TFloat32](/jvm/api_docs/java/org/tensorflow/types/TFloat32)\\\u003e\\\u003e learningRates, String config) Factory method to create a class wrapping a new SendTPUEmbeddingGradients operation. |\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: \"SendTPUEmbeddingGradients\"\n\nPublic Methods\n--------------\n\n#### public static [SendTPUEmbeddingGradients](/jvm/api_docs/java/org/tensorflow/op/tpu/SendTPUEmbeddingGradients)\n**create**\n([Scope](/jvm/api_docs/java/org/tensorflow/op/Scope) scope, Iterable\\\u003c[Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003c[TFloat32](/jvm/api_docs/java/org/tensorflow/types/TFloat32)\\\u003e\\\u003e inputs, Iterable\\\u003c[Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003c[TFloat32](/jvm/api_docs/java/org/tensorflow/types/TFloat32)\\\u003e\\\u003e learningRates, String config)\n\nFactory method to create a class wrapping a new SendTPUEmbeddingGradients operation. \n\n##### Parameters\n\n| scope | current scope |\n| inputs | A TensorList of gradients with which to update embedding tables. This argument has the same length and shapes as the return value of RecvTPUEmbeddingActivations, but contains gradients of the model's loss with respect to the embedding activations. The embedding tables are updated from these gradients via the optimizer specified in the TPU embedding configuration given to tpu.initialize_system. |\n| learningRates | A TensorList of float32 scalars, one for each dynamic learning rate tag: see the comments in //third_party/tensorflow/core/protobuf/tpu/optimization_parameters.proto. Multiple tables can share the same dynamic learning rate tag as specified in the configuration. If the learning rates for all tables are constant, this list should be empty. |\n| config | Serialized TPUEmbeddingConfiguration proto. |\n|---------------|--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|\n\n##### Returns\n\n- a new instance of SendTPUEmbeddingGradients"]]
