Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[QNN] Doc fix on convolution and dequantize #4799

Merged
merged 3 commits into from
Feb 3, 2020
Merged

[QNN] Doc fix on convolution and dequantize #4799

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
cc12785
QNN doc fix on conv and dequantize
masahi Feb 1, 2020
ee065e0
fix param name in tflite frontend
masahi Feb 1, 2020
a62fa77
make different fix
masahi Feb 1, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 6 additions & 6 deletions python/tvm/relay/qnn/op/qnn.py
View file
Open in desktop
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