Class LoadAllTPUEmbeddingParameters.Inputs

java.lang.Object

org.tensorflow.op.RawOpInputs<LoadAllTPUEmbeddingParameters>

org.tensorflow.op.tpu.LoadAllTPUEmbeddingParameters.Inputs

Enclosing class:

LoadAllTPUEmbeddingParameters

public static class LoadAllTPUEmbeddingParameters.Inputs
extends RawOpInputs<LoadAllTPUEmbeddingParameters>

Field Summary

Fields

Modifier and Type	Field	Description
final Iterable<Operand<TFloat32>>	auxiliary1	A list of tensors, one for each embedding table, containing the initial values of the first auxiliary optimization parameter to use in embedding training loop updates.
final Iterable<Operand<TFloat32>>	auxiliary2	A list of tensors, one for each embedding table, containing the initial values of the second auxiliary optimization parameter to use in embedding training loop updates.
final Iterable<Operand<TFloat32>>	auxiliary3	A list of tensors, one for each embedding table, containing the initial values of the third auxiliary optimization parameter to use in embedding training loop updates.
final Iterable<Operand<TFloat32>>	auxiliary4	A list of tensors, one for each embedding table, containing the initial values of the second auxiliary optimization parameter to use in embedding training loop updates.
final Iterable<Operand<TFloat32>>	auxiliary5	A list of tensors, one for each embedding table, containing the initial values of the third auxiliary optimization parameter to use in embedding training loop updates.
final Iterable<Operand<TFloat32>>	auxiliary6	A list of tensors, one for each embedding table, containing the initial values of the second auxiliary optimization parameter to use in embedding training loop updates.
final Iterable<Operand<TFloat32>>	auxiliary7	A list of tensors, one for each embedding table, containing the initial values of the third auxiliary optimization parameter to use in embedding training loop updates.
final String	config	An TPUEmbeddingConfiguration proto describing the table parameters being loaded, serialized to a string.
final long	numShards	Number of shards into which the embedding tables are divided.
final Iterable<Operand<TFloat32>>	parameters	A list of tensors, one for each embedding table, containing the initial embedding table parameters to use in embedding lookups.
final long	shardId	Identifier of shard for this operation.

Constructor Summary

Constructors

Constructor	Description
Inputs(GraphOperation op)

Method Summary

Methods inherited from class RawOpInputs

attributeMetadata, attributeNames, attributes, attributeValue, attributeValues, equals, getOutputs, hashCode, toString

Methods inherited from class Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Field Details

parameters

public final Iterable<Operand<TFloat32>> parameters

A list of tensors, one for each embedding table, containing the initial embedding table parameters to use in embedding lookups.

auxiliary1

public final Iterable<Operand<TFloat32>> auxiliary1

A list of tensors, one for each embedding table, containing the initial values of the first auxiliary optimization parameter to use in embedding training loop updates. The shape of each entry is ignored (and thus can be empty) for those tables whose optimization algorithms do not have at least one auxiliary parameter.

auxiliary2

public final Iterable<Operand<TFloat32>> auxiliary2

A list of tensors, one for each embedding table, containing the initial values of the second auxiliary optimization parameter to use in embedding training loop updates. The shape of each entry is ignored (and thus can be empty) for those tables whose optimization algorithms do not have at least two auxiliary

auxiliary3

public final Iterable<Operand<TFloat32>> auxiliary3

A list of tensors, one for each embedding table, containing the initial values of the third auxiliary optimization parameter to use in embedding training loop updates. The shape of each entry is ignored (and thus can be empty) for those tables whose optimization algorithms do not have three auxiliary parameters.

auxiliary4

public final Iterable<Operand<TFloat32>> auxiliary4

A list of tensors, one for each embedding table, containing the initial values of the second auxiliary optimization parameter to use in embedding training loop updates. The shape of each entry is ignored (and thus can be empty) for those tables whose optimization algorithms do not have at least four auxiliary

auxiliary5

public final Iterable<Operand<TFloat32>> auxiliary5

A list of tensors, one for each embedding table, containing the initial values of the third auxiliary optimization parameter to use in embedding training loop updates. The shape of each entry is ignored (and thus can be empty) for those tables whose optimization algorithms do not have five auxiliary parameters.

auxiliary6

public final Iterable<Operand<TFloat32>> auxiliary6

A list of tensors, one for each embedding table, containing the initial values of the second auxiliary optimization parameter to use in embedding training loop updates. The shape of each entry is ignored (and thus can be empty) for those tables whose optimization algorithms do not have at least six auxiliary

auxiliary7

public final Iterable<Operand<TFloat32>> auxiliary7

A list of tensors, one for each embedding table, containing the initial values of the third auxiliary optimization parameter to use in embedding training loop updates. The shape of each entry is ignored (and thus can be empty) for those tables whose optimization algorithms do not have sevan auxiliary parameters.

config

public final String config

An TPUEmbeddingConfiguration proto describing the table parameters being loaded, serialized to a string.

numShards

public final long numShards

Number of shards into which the embedding tables are divided.

shardId

public final long shardId

Identifier of shard for this operation.

Constructor Details

Inputs

public Inputs(GraphOperation op)