Unverified Commit 9963cf38 by masahi Committed by GitHub

[QNN] Doc fix on convolution and dequantize (#4799)

* QNN doc fix on conv and dequantize

* fix param name in tflite frontend

* make different fix
parent 396095a3
...@@ -131,9 +131,9 @@ def dequantize(data, ...@@ -131,9 +131,9 @@ def dequantize(data,
data : tvm.relay.Expr data : tvm.relay.Expr
The input tensor to be dequantized. Can be of type [int8, uint8]. The input tensor to be dequantized. Can be of type [int8, uint8].
input_zero_point : tvm.relay.Expr input_zero_point : tvm.relay.Expr
The output zero_point. The input zero_point.
input_scale : tvm.relay.Expr input_scale : tvm.relay.Expr
The output scale. The input scale.
Returns Returns
------- -------
result : tvm.relay.Expr result : tvm.relay.Expr
...@@ -201,12 +201,12 @@ def conv2d(data, ...@@ -201,12 +201,12 @@ def conv2d(data,
kernel_zero_point, kernel_zero_point,
input_scale, input_scale,
kernel_scale, kernel_scale,
kernel_size,
strides=(1, 1), strides=(1, 1),
padding=(0, 0), padding=(0, 0),
dilation=(1, 1), dilation=(1, 1),
groups=1, groups=1,
channels=None, channels=None,
kernel_size=None,
data_layout="NCHW", data_layout="NCHW",
kernel_layout="OIHW", kernel_layout="OIHW",
out_layout="", out_layout="",
...@@ -244,6 +244,9 @@ def conv2d(data, ...@@ -244,6 +244,9 @@ def conv2d(data,
needed in the pass pipeline after qnn.conv2d is lowered to the 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. 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 strides : tuple of int, optional
The strides of convolution. The strides of convolution.
...@@ -259,9 +262,6 @@ def conv2d(data, ...@@ -259,9 +262,6 @@ def conv2d(data,
channels : int, optional channels : int, optional
Number of output channels of this convolution. Number of output channels of this convolution.
kernel_size : tuple of int, optional
The spatial of the convolution kernel.
data_layout : str, optional data_layout : str, optional
Layout of the input. Layout of the input.
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment