V - data type for out() output@Operator public final class QuantizedMatMul<V> extends PrimitiveOp
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).
| Modifier and Type | Class and Description |
|---|---|
static class |
QuantizedMatMul.Options
Optional attributes for
QuantizedMatMul |
operation| Modifier and Type | Method and Description |
|---|---|
static <V,T,U,W> QuantizedMatMul<V> |
create(Scope scope,
Operand<T> a,
Operand<U> b,
Operand<Float> minA,
Operand<Float> maxA,
Operand<Float> minB,
Operand<Float> maxB,
Class<V> Toutput,
Class<W> Tactivation,
QuantizedMatMul.Options... options)
Factory method to create a class to wrap a new QuantizedMatMul operation to the graph.
|
Output<Float> |
maxOut()
The float value that the highest quantized output value represents.
|
Output<Float> |
minOut()
The float value that the lowest quantized output value represents.
|
Output<V> |
out() |
static QuantizedMatMul.Options |
transposeA(Boolean transposeA) |
static QuantizedMatMul.Options |
transposeB(Boolean transposeB) |
equals, hashCode, toStringpublic static <V,T,U,W> QuantizedMatMul<V> create(Scope scope, Operand<T> a, Operand<U> b, Operand<Float> minA, Operand<Float> maxA, Operand<Float> minB, Operand<Float> maxB, Class<V> Toutput, Class<W> Tactivation, QuantizedMatMul.Options... options)
scope - current graph scopea - Must be a two-dimensional tensor.b - Must be a two-dimensional tensor.minA - The float value that the lowest quantized `a` value represents.maxA - The float value that the highest quantized `a` value represents.minB - The float value that the lowest quantized `b` value represents.maxB - The float value that the highest quantized `b` value represents.Toutput - Tactivation - The type of output produced by activation function
following this operation.options - carries optional attributes valuespublic static QuantizedMatMul.Options transposeA(Boolean transposeA)
transposeA - If true, `a` is transposed before multiplication.public static QuantizedMatMul.Options transposeB(Boolean transposeB)
transposeB - If true, `b` is transposed before multiplication.public Output<Float> minOut()
Copyright © 2015–2019. All rights reserved.