Mxnet parser for Qnn dialect (apache#4714)

* - Additional util methods needed for mxnet frontend for qnn dialect. * - Fixing call to quantize. * [QNN] MxNet-MKLDNN parser support for QNN * [QNN] Relax conv check. * - Merge from origin * [QNN] Channel wise changes * [QNN] Dense changes * Dense fix for QNN ops. * - Removed non-mkl code from utils. - Small refactoring - Remove "with_sum" from conv - Simplified code * - Fixing ring buffer name. * - Fixing pylint issues. * - Fixing lint - Removing redundant commented code. * - Adding test cases - Removing unused methods. * [WIP] end to end test case for mxnet qnn parser * Changes to parse large CV models. * Pylint issues. * Fix Conv2D with sum and quantized pooling. * Reverting the changes made for mxnet-mkldnn test cases. Because of apache#4753, mxnet could not be updated to mxnet-mkldnn. Co-authored-by: Animesh Jain <[email protected]>
alexwong · Feb 26, 2020 · 6c4f3cd · 6c4f3cd
1 parent f8c9528
commit 6c4f3cd
Show file tree

Hide file tree

Showing 5 changed files with 1,061 additions and 158 deletions.
diff --git a/python/tvm/relay/frontend/__init__.py b/python/tvm/relay/frontend/__init__.py
@@ -25,6 +25,10 @@
 
 from .mxnet import from_mxnet
 from .mxnet_qnn_op_utils import dequantize_mxnet_min_max
+from .mxnet_qnn_op_utils import quantize_mxnet_min_max
+from .mxnet_qnn_op_utils import get_mkldnn_int8_scale
+from .mxnet_qnn_op_utils import get_mkldnn_uint8_scale
+from .mxnet_qnn_op_utils import quantize_conv_bias_mkldnn_from_var
 from .keras import from_keras
 from .onnx import from_onnx
 from .tflite import from_tflite