Class XlaSparseDenseMatmulGradWithSgdAndCsrInput

java.lang.Object

org.tensorflow.op.RawOp

org.tensorflow.op.xla.XlaSparseDenseMatmulGradWithSgdAndCsrInput

All Implemented Interfaces:

Shaped, Op, Operand<TFloat32>

@Operator(group="xla")
public final class XlaSparseDenseMatmulGradWithSgdAndCsrInput
extends RawOp
implements Operand<TFloat32>

The XlaSparseDenseMatmulGradWithSgdAndCsrInput operation

Nested Class Summary

Nested Classes

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

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
XlaSparseDenseMatmulGradWithSgdAndCsrInput(Operation operation)

Method Summary

Modifier and Type	Method	Description
Output<TFloat32>	asOutput()	Returns the symbolic handle of the tensor.
static XlaSparseDenseMatmulGradWithSgdAndCsrInput.Options	clipWeightMax(Float clipWeightMax)	Sets the clipWeightMax option.
static XlaSparseDenseMatmulGradWithSgdAndCsrInput.Options	clipWeightMin(Float clipWeightMin)	Sets the clipWeightMin option.
static XlaSparseDenseMatmulGradWithSgdAndCsrInput	create(Scope scope, Operand<TInt32> rowPointers, Operand<TInt32> sortedSampleIds, Operand<TInt32> sortedTokenIds, Operand<TFloat32> sortedGains, Operand<TFloat32> activationGradients, Operand<TFloat32> learningRate, Operand<TFloat32> embeddingTable, Operand<TInt32> numMinibatchesPerPhysicalSparseCore, String tableName, XlaSparseDenseMatmulGradWithSgdAndCsrInput.Options... options)	Factory method to create a class wrapping a new XlaSparseDenseMatmulGradWithSgdAndCsrInput operation.
Output<TFloat32>	updatedEmbeddingTable()	Gets updatedEmbeddingTable.

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, op

Methods inherited from interface Operand

asTensor, shape, type

Methods inherited from interface Shaped

rank, size

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

XlaSparseDenseMatmulGradWithSgdAndCsrInput

public XlaSparseDenseMatmulGradWithSgdAndCsrInput(Operation operation)

Method Details

create

@Endpoint(describeByClass=true)
public static XlaSparseDenseMatmulGradWithSgdAndCsrInput create(Scope scope,
 Operand<TInt32> rowPointers,
 Operand<TInt32> sortedSampleIds,
 Operand<TInt32> sortedTokenIds,
 Operand<TFloat32> sortedGains,
 Operand<TFloat32> activationGradients,
 Operand<TFloat32> learningRate,
 Operand<TFloat32> embeddingTable,
 Operand<TInt32> numMinibatchesPerPhysicalSparseCore,
 String tableName,
 XlaSparseDenseMatmulGradWithSgdAndCsrInput.Options... options)

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

Parameters:

scope - current scope

rowPointers - The rowPointers value

sortedSampleIds - The sortedSampleIds value

sortedTokenIds - The sortedTokenIds value

sortedGains - The sortedGains value

activationGradients - The activationGradients value

learningRate - The learningRate value

embeddingTable - The embeddingTable value

numMinibatchesPerPhysicalSparseCore - The numMinibatchesPerPhysicalSparseCore value

tableName - The value of the tableName attribute

options - carries optional attribute values

Returns:

a new instance of XlaSparseDenseMatmulGradWithSgdAndCsrInput

clipWeightMin

public static XlaSparseDenseMatmulGradWithSgdAndCsrInput.Options clipWeightMin(Float clipWeightMin)

Sets the clipWeightMin option.

Parameters:

clipWeightMin - the clipWeightMin option

Returns:

this Options instance.

clipWeightMax

public static XlaSparseDenseMatmulGradWithSgdAndCsrInput.Options clipWeightMax(Float clipWeightMax)

Sets the clipWeightMax option.

Parameters:

clipWeightMax - the clipWeightMax option

Returns:

this Options instance.

updatedEmbeddingTable

public Output<TFloat32> updatedEmbeddingTable()

Gets updatedEmbeddingTable.

Returns:

updatedEmbeddingTable.

asOutput

public Output<TFloat32> asOutput()

Description copied from interface: Operand

Returns the symbolic handle of the tensor.

Inputs to TensorFlow operations are outputs of another TensorFlow operation. This method is used to obtain a symbolic handle that represents the computation of the input.

Specified by:

asOutput in interface Operand<TFloat32>

See Also:

• OperationBuilder.addInput(Output)