From 42dd7453a8a9d3976bd18b042b688bd237fa8375 Mon Sep 17 00:00:00 2001 From: Animesh Jain Date: Mon, 30 Sep 2019 05:41:01 +0000 Subject: [PATCH] [QNN] Renaming dense operator. --- python/tvm/relay/qnn/op/qnn.py | 12 ++++++------ tests/python/relay/test_qnn_dense.py | 2 +- 2 files changed, 7 insertions(+), 7 deletions(-) diff --git a/python/tvm/relay/qnn/op/qnn.py b/python/tvm/relay/qnn/op/qnn.py index 878a3a72b01e..ed443abb5293 100644 --- a/python/tvm/relay/qnn/op/qnn.py +++ b/python/tvm/relay/qnn/op/qnn.py @@ -313,12 +313,12 @@ def add(lhs, output_scale, output_zero_point) -def quantized_dense(data, - weight, - input_zero_point, - kernel_zero_point, - units=None, - out_dtype="int32"): +def dense(data, + weight, + input_zero_point, + kernel_zero_point, + units=None, + out_dtype="int32"): """Qnn Dense operator. Applies a quantized linear transformation diff --git a/tests/python/relay/test_qnn_dense.py b/tests/python/relay/test_qnn_dense.py index 2d14593b331f..f1e0767aff2e 100644 --- a/tests/python/relay/test_qnn_dense.py +++ b/tests/python/relay/test_qnn_dense.py @@ -153,7 +153,7 @@ def qnn_dense_driver(test_configuration): quantized_kernel = relay.var(quantized_kernel_name, shape=test_configuration['kernel_shape'], dtype=in_dtype) - mod = relay.qnn.op.quantized_dense( + mod = relay.qnn.op.dense( quantized_data, quantized_kernel, test_configuration['input_zero_point'],