XlaSparseDenseMatmulWithCsrInput

public final class XlaSparseDenseMatmulWithCsrInput

Public Methods

Output<Float>
activations()
Output<Float>
asOutput()
Returns the symbolic handle of a tensor.
static XlaSparseDenseMatmulWithCsrInput
create(Scope scope, Operand<Integer> rowPointers, Operand<Integer> sortedSampleIds, Operand<Integer> sortedTokenIds, Operand<Float> sortedGains, Operand<Float> embeddingTable, Operand<Integer> numMinibatchesPerPhysicalSparseCore, Long inputSize, Float quantizationConfigLow, Float quantizationConfigHigh, Long quantizationConfigNumBuckets, String tableName)
Factory method to create a class wrapping a new XlaSparseDenseMatmulWithCsrInput operation.

Inherited Methods

org.tensorflow.op.PrimitiveOp
final boolean
equals(Object obj)
final int
hashCode()
Operation
op()
Returns the underlying Operation
final String
toString()
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()
org.tensorflow.Operand
abstract Output<Float>
asOutput()
Returns the symbolic handle of a tensor.

Public Methods

public Output<Float> activations ()

public Output<Float> asOutput ()

Returns the symbolic handle of a tensor.

Inputs to TensorFlow operations are outputs of another TensorFlow operation. This method is used to obtain a symbolic handle that represents the computation of the input.

public static XlaSparseDenseMatmulWithCsrInput create (Scope scope, Operand<Integer> rowPointers, Operand<Integer> sortedSampleIds, Operand<Integer> sortedTokenIds, Operand<Float> sortedGains, Operand<Float> embeddingTable, Operand<Integer> numMinibatchesPerPhysicalSparseCore, Long inputSize, Float quantizationConfigLow, Float quantizationConfigHigh, Long quantizationConfigNumBuckets, String tableName)

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

Parameters
scope current scope
Returns
  • a new instance of XlaSparseDenseMatmulWithCsrInput