Perform a quantized matrix multiplication of a by the matrix b.
tf.raw_ops.QuantizedMatMul(
a,
b,
min_a,
max_a,
min_b,
max_b,
Toutput=tf.dtypes.qint32,
transpose_a=False,
transpose_b=False,
Tactivation=tf.dtypes.quint8,
name=None
)
The inputs must be two-dimensional matrices and the inner dimension of
a (after being transposed if transpose_a is non-zero) must match the
outer dimension of b (after being transposed if transposed_b is
non-zero).
Args | |
|---|---|
a
|
A Tensor. Must be one of the following types: qint8, quint8, qint32, qint16, quint16.
Must be a two-dimensional tensor.
|
b
|
A Tensor. Must be one of the following types: qint8, quint8, qint32, qint16, quint16.
Must be a two-dimensional tensor.
|
min_a
|
A Tensor of type float32.
The float value that the lowest quantized a value represents.
|
max_a
|
A Tensor of type float32.
The float value that the highest quantized a value represents.
|
min_b
|
A Tensor of type float32.
The float value that the lowest quantized b value represents.
|
max_b
|
A Tensor of type float32.
The float value that the highest quantized b value represents.
|
Toutput
|
An optional tf.DType from: tf.qint8, tf.quint8, tf.qint32, tf.qint16, tf.quint16. Defaults to tf.qint32.
|
transpose_a
|
An optional bool. Defaults to False.
If true, a is transposed before multiplication.
|
transpose_b
|
An optional bool. Defaults to False.
If true, b is transposed before multiplication.
|
Tactivation
|
An optional tf.DType from: tf.qint8, tf.quint8, tf.qint32, tf.qint16, tf.quint16. Defaults to tf.quint8.
The type of output produced by activation function
following this operation.
|
name
|
A name for the operation (optional). |
Returns | |
|---|---|
A tuple of Tensor objects (out, min_out, max_out).
|
|
out
|
A Tensor of type Toutput.
|
min_out
|
A Tensor of type float32.
|
max_out
|
A Tensor of type float32.
|