Class RetrieveTPUEmbeddingFTRLParameters

java.lang.Object

org.tensorflow.op.RawOp

org.tensorflow.op.tpu.RetrieveTPUEmbeddingFTRLParameters

All Implemented Interfaces:

Op

@Operator(group="tpu")
public final class RetrieveTPUEmbeddingFTRLParameters
extends RawOp

Retrieve FTRL embedding parameters. An op that retrieves optimization parameters from embedding to host memory. Must be preceded by a ConfigureTPUEmbeddingHost op that sets up the correct embedding table configuration. For example, this op is used to retrieve updated parameters before saving a checkpoint.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	RetrieveTPUEmbeddingFTRLParameters.Inputs
static class	RetrieveTPUEmbeddingFTRLParameters.Options	Optional attributes for RetrieveTPUEmbeddingFTRLParameters

Field Summary

Fields

Modifier and Type	Field	Description
static final String	OP_NAME	The name of this op, as known by TensorFlow core engine

Fields inherited from class RawOp

operation

Constructor Summary

Constructors

Constructor	Description
RetrieveTPUEmbeddingFTRLParameters(Operation operation)

Method Summary

Modifier and Type	Method	Description
Output<TFloat32>	accumulators()	Gets accumulators.
static RetrieveTPUEmbeddingFTRLParameters.Options	config(String config)	Sets the config option.
static RetrieveTPUEmbeddingFTRLParameters	create(Scope scope, Long numShards, Long shardId, RetrieveTPUEmbeddingFTRLParameters.Options... options)	Factory method to create a class wrapping a new RetrieveTPUEmbeddingFTRLParameters operation.
Output<TFloat32>	linears()	Gets linears.
Output<TFloat32>	parameters()	Gets parameters.
static RetrieveTPUEmbeddingFTRLParameters.Options	tableId(Long tableId)	Sets the tableId option.
static RetrieveTPUEmbeddingFTRLParameters.Options	tableName(String tableName)	Sets the tableName option.

Methods inherited from class RawOp

equals, hashCode, op, toString

Methods inherited from class Object

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

Methods inherited from interface Op

env

Field Details

OP_NAME

public static final String OP_NAME

The name of this op, as known by TensorFlow core engine

See Also:

• Constant Field Values

Constructor Details

RetrieveTPUEmbeddingFTRLParameters

public RetrieveTPUEmbeddingFTRLParameters(Operation operation)

Method Details

create

@Endpoint(describeByClass=true)
public static RetrieveTPUEmbeddingFTRLParameters create(Scope scope,
 Long numShards,
 Long shardId,
 RetrieveTPUEmbeddingFTRLParameters.Options... options)

Factory method to create a class wrapping a new RetrieveTPUEmbeddingFTRLParameters operation.

Parameters:

scope - current scope

numShards - The value of the numShards attribute

shardId - The value of the shardId attribute

options - carries optional attribute values

Returns:

a new instance of RetrieveTPUEmbeddingFTRLParameters

tableId

public static RetrieveTPUEmbeddingFTRLParameters.Options tableId(Long tableId)

Sets the tableId option.

Parameters:

tableId - the tableId option

Returns:

this Options instance.

tableName

public static RetrieveTPUEmbeddingFTRLParameters.Options tableName(String tableName)

Sets the tableName option.

Parameters:

tableName - the tableName option

Returns:

this Options instance.

config

public static RetrieveTPUEmbeddingFTRLParameters.Options config(String config)

Sets the config option.

Parameters:

config - the config option

Returns:

this Options instance.

parameters

public Output<TFloat32> parameters()

Gets parameters. Parameter parameters updated by the FTRL optimization algorithm.

Returns:

parameters.

accumulators

public Output<TFloat32> accumulators()

Gets accumulators. Parameter accumulators updated by the FTRL optimization algorithm.

Returns:

accumulators.

linears

public Output<TFloat32> linears()

Gets linears. Parameter linears updated by the FTRL optimization algorithm.

Returns:

linears.