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

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)