Class XlaSparseDenseMatmul.Inputs

java.lang.Object

org.tensorflow.op.RawOpInputs<XlaSparseDenseMatmul>

org.tensorflow.op.xla.XlaSparseDenseMatmul.Inputs

Enclosing class:

XlaSparseDenseMatmul

public static class XlaSparseDenseMatmul.Inputs
extends RawOpInputs<XlaSparseDenseMatmul>

Field Summary

Fields

Modifier and Type	Field	Description
final Operand<? extends TType>	colIds	The colIds input
final Operand<TFloat32>	embeddingTable	The embeddingTable input
final long	inputSize	The inputSize attribute
final long	maxIdsPerPartition	The maxIdsPerPartition attribute
final long	maxUniqueIdsPerPartition	The maxUniqueIdsPerPartition attribute
final Operand<? extends TType>	offsets	The offsets input
final Operand<TInt32>	rowIds	The rowIds input
final Operand<TFloat32>	values	The values input

Constructor Summary

Constructors

Constructor	Description
Inputs(GraphOperation op)

Method Summary

Methods inherited from class RawOpInputs

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

Modifier and Type	Method	Description
AttributeMetadata	attributeMetadata(String name)	Get the metadata for an attribute
Set<String>	attributeNames()	Get the names of this op's attributes
OperationAttributeInspector	attributes()	Get an inspector for the operation's attributes.
AttrValue	attributeValue(String name)	Get the value of an attribute as an AttrValue proto.
Map<String, AttrValue>	attributeValues()	Get all attribute value protos
final boolean	equals(Object obj)
XlaSparseDenseMatmul	getOutputs()	The outputs of this operation.
final int	hashCode()
final String	toString()

Methods inherited from class Object

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

Field Details

rowIds

public final Operand<TInt32> rowIds

The rowIds input

colIds

public final Operand<? extends TType> colIds

The colIds input

values

public final Operand<TFloat32> values

The values input

offsets

public final Operand<? extends TType> offsets

The offsets input

embeddingTable

public final Operand<TFloat32> embeddingTable

The embeddingTable input

maxIdsPerPartition

public final long maxIdsPerPartition

The maxIdsPerPartition attribute

maxUniqueIdsPerPartition

public final long maxUniqueIdsPerPartition

The maxUniqueIdsPerPartition attribute

inputSize

public final long inputSize

The inputSize attribute

Constructor Details

Inputs

public Inputs(GraphOperation op)