This is an automated email from the ASF dual-hosted git repository.
patriczhao pushed a change to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git.
from 649429d Disable flaky test in test_amp_conversion (#16031)
add 3f7b6ee Improve quantization flow (#15961)
No new revisions were added by this update.
Summary of changes:
example/quantization/README.md | 23 +-
example/quantization/imagenet_gen_qsym.py | 17 +-
example/quantization/imagenet_gen_qsym_mkldnn.py | 3 +-
example/ssd/quantization.py | 11 +-
include/mxnet/c_api.h | 7 +-
include/mxnet/op_attr_types.h | 30 ++
python/mxnet/contrib/quantization.py | 285 ++++++++-----------
src/c_api/c_api_symbolic.cc | 17 +-
src/common/utils.h | 10 +
src/executor/graph_executor.cc | 15 +-
.../mkldnn/mkldnn_flatten-inl.h} | 30 +-
src/operator/nn/mkldnn/mkldnn_flatten.cc | 10 +-
.../np_init_op.cu => quantization/calibrate-inl.h} | 35 +--
src/operator/quantization/calibrate.cc | 215 ++++++++++++++
.../quantization/mkldnn/mkldnn_quantize_v2-inl.h | 6 +-
.../mkldnn/mkldnn_quantized_elemwise_add.cc | 4 +-
.../mkldnn/mkldnn_quantized_flatten.cc | 61 ++++
src/operator/quantization/quantization_utils.h | 4 +-
src/operator/quantization/quantize_graph_pass.cc | 315 +++++++++++++--------
src/operator/quantization/quantize_v2.cc | 3 +
src/operator/quantization/quantized_batch_norm.cc | 3 +
src/operator/quantization/quantized_conv.cc | 3 +
.../quantization/quantized_fully_connected.cc | 3 +
src/operator/quantization/requantize.cc | 3 +
src/operator/subgraph/mkldnn/mkldnn_conv.cc | 3 +
src/operator/subgraph/mkldnn/mkldnn_fc.cc | 3 +
tests/python/mkl/test_subgraph.py | 9 +-
tests/python/quantization/test_quantization.py | 53 +++-
tests/python/unittest/test_operator.py | 5 +-
29 files changed, 776 insertions(+), 410 deletions(-)
copy src/operator/{quantization/mkldnn/mkldnn_quantized_ops-inl.h =>
nn/mkldnn/mkldnn_flatten-inl.h} (54%)
copy src/operator/{numpy/np_init_op.cu => quantization/calibrate-inl.h} (60%)
create mode 100644 src/operator/quantization/calibrate.cc
create mode 100644 src/operator/quantization/mkldnn/mkldnn_quantized_flatten.cc
