diff --git a/python/tvm/relay/qnn/op/qnn.py b/python/tvm/relay/qnn/op/qnn.py index c79c99c80d442..f32eaa14ad86d 100644 --- a/python/tvm/relay/qnn/op/qnn.py +++ b/python/tvm/relay/qnn/op/qnn.py @@ -221,6 +221,9 @@ def conv2d(data, input_zero_point: int The zero point of the data distribution. + kernel_zero_point: int + The zero point of the quantized_kernel distribution. + input_scale: float The scale for the input tensor. The scale for the input tensor is stored purely for convenience here. See more commentary below. @@ -231,9 +234,6 @@ 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_zero_point: int - The zero point of the quantized_kernel distribution. - strides : tuple of int, optional The strides of convolution.