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