Class XlaSparseDenseMatmulCustomCombinerOnTcGradWithCsrInput

java.lang.Object

org.tensorflow.op.RawOp

org.tensorflow.op.xla.XlaSparseDenseMatmulCustomCombinerOnTcGradWithCsrInput

All Implemented Interfaces:

Op

@Operator(group="xla")
public final class XlaSparseDenseMatmulCustomCombinerOnTcGradWithCsrInput
extends RawOp

The XlaSparseDenseMatmulCustomCombinerOnTcGradWithCsrInput operation

Nested Class Summary

Nested Classes

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

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

Modifier and Type	Field	Description
protected final Operation	operation

Constructor Summary

Constructors

Constructor	Description
XlaSparseDenseMatmulCustomCombinerOnTcGradWithCsrInput(Operation operation)

Method Summary

Modifier and Type	Method	Description
static XlaSparseDenseMatmulCustomCombinerOnTcGradWithCsrInput	create(Scope scope, Operand<TInt32> rowPointers, Operand<TInt32> sortedSampleIds, Operand<TInt32> sortedTokenIds, Operand<TInt32> sortedPosIds, Operand<TFloat32> sortedGains, Operand<TFloat32> weights, Operand<TInt32> preservedValencies, Operand<TFloat32> preservedVectors, Operand<TFloat32> preservedWeights, Operand<TFloat32> activationGradients, Iterable<Operand<TFloat32>> tables, Iterable<Operand<TFloat32>> hyperparameters, Operand<TFloat32> combinerWeightsLearningRate, Long maxValency, Long numWeights, ConcreteFunction combinerTableVjpComputation, ConcreteFunction combinerWeightsVjpComputation, ConcreteFunction optimizerCustomComputation, String tableName, XlaSparseDenseMatmulCustomCombinerOnTcGradWithCsrInput.Options... options)	Factory method to create a class wrapping a new XlaSparseDenseMatmulCustomCombinerOnTcGradWithCsrInput operation.
static XlaSparseDenseMatmulCustomCombinerOnTcGradWithCsrInput.Options	numSparsecoresPerDevice(Long numSparsecoresPerDevice)	Sets the numSparsecoresPerDevice option.
List<Output<TFloat32>>	updatedTables()	Gets updatedTables.
Output<TFloat32>	updatedWeights()	Gets updatedWeights.

Methods inherited from class RawOp

equals, hashCode, op, toString

Modifier and Type	Method	Description
final boolean	equals(Object obj)
final int	hashCode()
Operation	op()	Return this unit of computation as a single Operation.
final String	toString()

Methods inherited from class Object

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

Methods inherited from interface Op

env

Modifier and Type	Method	Description
default ExecutionEnvironment	env()	Return the execution environment this op was created in.

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

XlaSparseDenseMatmulCustomCombinerOnTcGradWithCsrInput

public XlaSparseDenseMatmulCustomCombinerOnTcGradWithCsrInput(Operation operation)

Method Details

create

@Endpoint(describeByClass=true)
public static XlaSparseDenseMatmulCustomCombinerOnTcGradWithCsrInput create(Scope scope,
 Operand<TInt32> rowPointers,
 Operand<TInt32> sortedSampleIds,
 Operand<TInt32> sortedTokenIds,
 Operand<TInt32> sortedPosIds,
 Operand<TFloat32> sortedGains,
 Operand<TFloat32> weights,
 Operand<TInt32> preservedValencies,
 Operand<TFloat32> preservedVectors,
 Operand<TFloat32> preservedWeights,
 Operand<TFloat32> activationGradients,
 Iterable<Operand<TFloat32>> tables,
 Iterable<Operand<TFloat32>> hyperparameters,
 Operand<TFloat32> combinerWeightsLearningRate,
 Long maxValency,
 Long numWeights,
 ConcreteFunction combinerTableVjpComputation,
 ConcreteFunction combinerWeightsVjpComputation,
 ConcreteFunction optimizerCustomComputation,
 String tableName,
 XlaSparseDenseMatmulCustomCombinerOnTcGradWithCsrInput.Options... options)

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

Parameters:

scope - current scope

rowPointers - The rowPointers value

sortedSampleIds - The sortedSampleIds value

sortedTokenIds - The sortedTokenIds value

sortedPosIds - The sortedPosIds value

sortedGains - The sortedGains value

weights - The weights value

preservedValencies - The preservedValencies value

preservedVectors - The preservedVectors value

preservedWeights - The preservedWeights value

activationGradients - The activationGradients value

tables - The tables value

hyperparameters - The hyperparameters value

combinerWeightsLearningRate - The combinerWeightsLearningRate value

maxValency - The value of the maxValency attribute

numWeights - The value of the numWeights attribute

combinerTableVjpComputation - The value of the combinerTableVjpComputation attribute

combinerWeightsVjpComputation - The value of the combinerWeightsVjpComputation attribute

optimizerCustomComputation - The value of the optimizerCustomComputation attribute

tableName - The value of the tableName attribute

options - carries optional attribute values

Returns:

a new instance of XlaSparseDenseMatmulCustomCombinerOnTcGradWithCsrInput

numSparsecoresPerDevice

public static XlaSparseDenseMatmulCustomCombinerOnTcGradWithCsrInput.Options numSparsecoresPerDevice(Long numSparsecoresPerDevice)

Sets the numSparsecoresPerDevice option.

Parameters:

numSparsecoresPerDevice - the numSparsecoresPerDevice option

Returns:

this Options instance.

updatedTables

public List<Output<TFloat32>> updatedTables()

Gets updatedTables.

Returns:

updatedTables.

updatedWeights

public Output<TFloat32> updatedWeights()

Gets updatedWeights.

Returns:

updatedWeights.