XlaSparseDenseMatmul

public final class XlaSparseDenseMatmul

Public Methods

Output<Float>	activations()
static XlaSparseDenseMatmul	create(Scope scope, Operand<Integer> rowIds, Operand<?> colIds, Operand<Float> values, Operand<?> offsets, Operand<Float> embeddingTable, Long maxIdsPerPartition, Long maxUniqueIdsPerPartition, Long inputSize) Factory method to create a class wrapping a new XlaSparseDenseMatmul operation.
Output<Integer>	rowPointers()
Output<Integer>	sortedEmbeddingIds()
Output<Float>	sortedGains()
Output<Integer>	sortedSampleIds()

Inherited Methods

From class org.tensorflow.op.PrimitiveOp

final boolean	equals(Object obj)
final int	hashCode()
Operation	op() Returns the underlying Operation
final String	toString()

From class java.lang.Object

boolean	equals(Object arg0)
final Class<?>	getClass()
int	hashCode()
final void	notify()
final void	notifyAll()
String	toString()
final void	wait(long arg0, int arg1)
final void	wait(long arg0)
final void	wait()

Public Methods

public Output<Float> activations ()

public static XlaSparseDenseMatmul create (Scope scope, Operand<Integer> rowIds, Operand<?> colIds, Operand<Float> values, Operand<?> offsets, Operand<Float> embeddingTable, Long maxIdsPerPartition, Long maxUniqueIdsPerPartition, Long inputSize)

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

Parameters

scope	current scope

Returns

• a new instance of XlaSparseDenseMatmul

public Output<Integer> rowPointers ()

public Output<Integer> sortedEmbeddingIds ()

public Output<Float> sortedGains ()

public Output<Integer> sortedSampleIds ()