Perform a quantized matrix multiplication of a
by the matrix b
.
View aliases
Compat aliases for migration
See Migration guide for more details.
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 Tensor of type float32 .
The float value that the lowest quantized a value represents.
|
max_
|
A Tensor of type float32 .
The float value that the highest quantized a value represents.
|
min_
|
A Tensor of type float32 .
The float value that the lowest quantized b value represents.
|
max_
|
A Tensor of type float32 .
The float value that the highest quantized b value represents.
|
Toutput
|
An optional tf.DType from: tf. . Defaults to tf.qint32 .
|
transpose_
|
An optional bool . Defaults to False .
If true, a is transposed before multiplication.
|
transpose_
|
An optional bool . Defaults to False .
If true, b is transposed before multiplication.
|
Tactivation
|
An optional tf.DType from: tf. . Defaults to tf.quint8 .
The type of output produced by activation function
following this operation.
|
name
|
A name for the operation (optional). |