Browse Source

!6857 support deconv quant op

Merge pull request !6857 from cjh9368/quant_same_datatype
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
6819d28fac
3 changed files with 3 additions and 1 deletions
  1. +1
    -1
      mindspore/lite/tools/common/node_util.cc
  2. +1
    -0
      mindspore/lite/tools/converter/quantizer/aware_quantizer.cc
  3. +1
    -0
      mindspore/lite/tools/converter/quantizer/calc_quant_param.cc

+ 1
- 1
mindspore/lite/tools/common/node_util.cc View File

@@ -79,7 +79,7 @@ static const std::vector<schema::PrimitiveType> 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<schema::PrimitiveType> needInsertOpList = {
schema::PrimitiveType_Eltwise, schema::PrimitiveType_Activation, schema::PrimitiveType_Concat,


+ 1
- 0
mindspore/lite/tools/converter/quantizer/aware_quantizer.cc View File

@@ -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;


+ 1
- 0
mindspore/lite/tools/converter/quantizer/calc_quant_param.cc View File

@@ -475,6 +475,7 @@ QuantParamCalcRegister::QuantParamCalcRegister() {
_registerMap[schema::PrimitiveType_Add] = std::make_shared<CalcAdd>();
_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;


Loading…
Cancel
Save