From ffc004395a4a7051a8aafad0c4aa4d20317fd692 Mon Sep 17 00:00:00 2001 From: cjh9368 Date: Thu, 24 Sep 2020 21:04:01 +0800 Subject: [PATCH] support deconv quant op --- mindspore/lite/tools/common/node_util.cc | 2 +- mindspore/lite/tools/converter/quantizer/aware_quantizer.cc | 1 + mindspore/lite/tools/converter/quantizer/calc_quant_param.cc | 1 + 3 files changed, 3 insertions(+), 1 deletion(-) diff --git a/mindspore/lite/tools/common/node_util.cc b/mindspore/lite/tools/common/node_util.cc index c0fd102a51..1ad5189f12 100644 --- a/mindspore/lite/tools/common/node_util.cc +++ b/mindspore/lite/tools/common/node_util.cc @@ -79,7 +79,7 @@ static const std::vector int8OpList = { schema::PrimitiveType_Squeeze, schema::PrimitiveType_Sub, schema::PrimitiveType_StridedSlice, schema::PrimitiveType_TopK, schema::PrimitiveType_Unsqueeze, schema::PrimitiveType_MatMul, - schema::PrimitiveType_Pad}; + schema::PrimitiveType_Pad, schema::PrimitiveType_DeConv2D}; static const std::vector needInsertOpList = { schema::PrimitiveType_Eltwise, schema::PrimitiveType_Activation, schema::PrimitiveType_Concat, diff --git a/mindspore/lite/tools/converter/quantizer/aware_quantizer.cc b/mindspore/lite/tools/converter/quantizer/aware_quantizer.cc index 468cc7a62a..3ec4cf5f6a 100644 --- a/mindspore/lite/tools/converter/quantizer/aware_quantizer.cc +++ b/mindspore/lite/tools/converter/quantizer/aware_quantizer.cc @@ -149,6 +149,7 @@ STATUS AwareQuantizer::DoQuantize() { STATUS status; if (GetCNodeTType(*node) == schema::PrimitiveType_Conv2D || GetCNodeTType(*node) == schema::PrimitiveType_DepthwiseConv2D || + GetCNodeTType(*node) == schema::PrimitiveType_DeConv2D || GetCNodeTType(*node) == schema::PrimitiveType_FullConnection || GetCNodeTType(*node) == schema::PrimitiveType_MatMul) { auto inputIndexes = node->inputIndex; diff --git a/mindspore/lite/tools/converter/quantizer/calc_quant_param.cc b/mindspore/lite/tools/converter/quantizer/calc_quant_param.cc index 8440053ba0..2370c5e5e8 100644 --- a/mindspore/lite/tools/converter/quantizer/calc_quant_param.cc +++ b/mindspore/lite/tools/converter/quantizer/calc_quant_param.cc @@ -475,6 +475,7 @@ QuantParamCalcRegister::QuantParamCalcRegister() { _registerMap[schema::PrimitiveType_Add] = std::make_shared(); _registerMap[schema::PrimitiveType_Mul] = commonCalcer; _registerMap[schema::PrimitiveType_Conv2D] = commonCalcer; + _registerMap[schema::PrimitiveType_DeConv2D] = commonCalcer; _registerMap[schema::PrimitiveType_DepthwiseConv2D] = commonCalcer; _registerMap[schema::PrimitiveType_Pooling] = linearCalcer; _registerMap[schema::PrimitiveType_Resize] = linearCalcer;