Skip to content

T
tic
Commit 0cd80478 by Animesh Jain Committed by Zhi
Options

[QNN] Renaming dense operator. (#4033)

parent d0fe532e
Showing with 7 additions and 7 deletions
python/tvm/relay/qnn/op/qnn.py
...@@ -313,12 +313,12 @@ def add(lhs, ...@@ -313,12 +313,12 @@ def add(lhs,
output_scale, output_zero_point) output_scale, output_zero_point)
def quantized_dense(data, def dense(data,
weight, weight,
input_zero_point, input_zero_point,
kernel_zero_point, kernel_zero_point,
units=None, units=None,
out_dtype="int32"): out_dtype="int32"):
"""Qnn Dense operator. """Qnn Dense operator.
Applies a quantized linear transformation Applies a quantized linear transformation
......
tests/python/relay/test_qnn_dense.py
...@@ -153,7 +153,7 @@ def qnn_dense_driver(test_configuration): ...@@ -153,7 +153,7 @@ def qnn_dense_driver(test_configuration):
quantized_kernel = relay.var(quantized_kernel_name, quantized_kernel = relay.var(quantized_kernel_name,
shape=test_configuration['kernel_shape'], shape=test_configuration['kernel_shape'],
dtype=in_dtype) dtype=in_dtype)
mod = relay.qnn.op.quantized_dense( mod = relay.qnn.op.dense(
quantized_data, quantized_data,
quantized_kernel, quantized_kernel,
test_configuration['input_zero_point'], test_configuration['input_zero_point'],
......
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