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

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 ()