Skip to content

Commit

Permalink
[QNN] Doc fix on convolution and dequantize (apache#4799)
Browse files Browse the repository at this point in the history
* QNN doc fix on conv and dequantize

* fix param name in tflite frontend

* make different fix
  • Loading branch information
masahi authored and zhiics committed Mar 2, 2020
1 parent 1e262ff commit b65e1d4
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions python/tvm/relay/qnn/op/qnn.py
Original file line number Diff line number Diff line change
Expand Up @@ -131,9 +131,9 @@ def dequantize(data,
data : tvm.relay.Expr
The input tensor to be dequantized. Can be of type [int8, uint8].
input_zero_point : tvm.relay.Expr
The output zero_point.
The input zero_point.
input_scale : tvm.relay.Expr
The output scale.
The input scale.
Returns
-------
result : tvm.relay.Expr
Expand Down Expand Up @@ -201,12 +201,12 @@ def conv2d(data,
kernel_zero_point,
input_scale,
kernel_scale,
kernel_size,
strides=(1, 1),
padding=(0, 0),
dilation=(1, 1),
groups=1,
channels=None,
kernel_size=None,
data_layout="NCHW",
kernel_layout="OIHW",
out_layout="",
Expand Down Expand Up @@ -244,6 +244,9 @@ def conv2d(data,
needed in the pass pipeline after qnn.conv2d is lowered to the
sequence of steps as in nn.conv2d. See also input_scale in Requantize.
kernel_size : tuple of int
The spatial width and height of the convolution kernel.
strides : tuple of int, optional
The strides of convolution.
Expand All @@ -259,9 +262,6 @@ def conv2d(data,
channels : int, optional
Number of output channels of this convolution.
kernel_size : tuple of int, optional
The spatial of the convolution kernel.
data_layout : str, optional
Layout of the input.
Expand Down

0 comments on commit b65e1d4

Please sign in to comment.