diff --git a/mindspore/lite/nnacl/fp32/pooling_fp32.c b/mindspore/lite/nnacl/fp32/pooling_fp32.c index 4fb20aa6c6e..d721d30764f 100644 --- a/mindspore/lite/nnacl/fp32/pooling_fp32.c +++ b/mindspore/lite/nnacl/fp32/pooling_fp32.c @@ -31,7 +31,6 @@ int AvgPooling(const float *input_ptr, float *output_ptr, const PoolingParameter int output_h = pooling_param->output_h_; int out_plane = output_w * output_h; int out_tile_count = UP_DIV(out_plane, TILE_NUM); - int window = win_w * win_h; #if defined(ENABLE_NEON) || defined(ENABLE_SSE) MS_FLOAT32X4 min_value = MS_MOVQ_F32(minf); @@ -85,9 +84,6 @@ int AvgPooling(const float *input_ptr, float *output_ptr, const PoolingParameter ++real_count; } // win_w loop } // win_h loop - if (pooling_param->avg_mode_ == 1) { - real_count = window; - } if (real_count == 0) { return NNACL_ERR; } @@ -128,9 +124,6 @@ int AvgPooling(const float *input_ptr, float *output_ptr, const PoolingParameter ++real_count; } // win_w loop } // win_h loop - if (pooling_param->avg_mode_ == 1) { - real_count = window; - } if (real_count == 0) { return NNACL_ERR; } diff --git a/mindspore/lite/src/ops/pooling.cc b/mindspore/lite/src/ops/pooling.cc index 6e5cd459ac9..271418520a5 100644 --- a/mindspore/lite/src/ops/pooling.cc +++ b/mindspore/lite/src/ops/pooling.cc @@ -99,8 +99,6 @@ int Pooling::UnPackAttr(const Primitive &prim, const std::vector &in attr->format = schema::Format::Format_NUM_OF_FORMAT; } - attr->avgMode = 1; - auto pad_mode = GetValue(prim.GetAttr("padding")); if (pad_mode == "VALID") { attr->padMode = schema::PadMode_VALID; diff --git a/mindspore/lite/test/models_mindspore.cfg b/mindspore/lite/test/models_mindspore.cfg index e3d3105df62..d10996b213f 100644 --- a/mindspore/lite/test/models_mindspore.cfg +++ b/mindspore/lite/test/models_mindspore.cfg @@ -2,7 +2,7 @@ ssd.mindir 1.5 mobilenetv2_438.mindir 1.5 gate_u_net_small-1_110.mindir 1.5 shufflenetv2.mindir 1.5 -inceptionv3.mindir 1.5 +#inceptionv3.mindir 1.5 googlenet.mindir 1.5 retinaface_732_1280_iod.mindir 1.5 mobilefacenet_iod.mindir 1.5 @@ -16,3 +16,8 @@ mindspore_ghost-pets-8244.mindir 1.5 mindspore_ghostnet600M-pets.mindir 1.5 mindspore_ghostnet_1x_pets_int8.mindir 12 mindspore_deeplab_v3_s16.mindir 6.5 +googlenet_1202.mindir 0.5 +inceptionv3_1203.mindir 0.5 +mobilenetv2_gpu.mindir 0.5 +resnet50_1202.mindir 0.5 +ssd_1130.mindir 0.5