Browse Source

fix format

tags/v1.0.0
sunsuodong 5 years ago
parent
commit
01766eef78
2 changed files with 3 additions and 3 deletions
  1. +2
    -2
      mindspore/lite/nnacl/fp32/pooling.c
  2. +1
    -1
      mindspore/lite/nnacl/fp32/pooling.h

+ 2
- 2
mindspore/lite/nnacl/fp32/pooling.c View File

@@ -19,7 +19,7 @@
#include "nnacl/errorcode.h"

int AvgPooling(const float *input_ptr, float *output_ptr, PoolingParameter *pooling_param, int task_id, float minf,
float maxf) {
float maxf) {
int stride_w = pooling_param->stride_w_;
int stride_h = pooling_param->stride_h_;
int pad_w = pooling_param->pad_l_;
@@ -137,7 +137,7 @@ int AvgPooling(const float *input_ptr, float *output_ptr, PoolingParameter *pool
real_count = window;
}
if (real_count == 0) {
return NNACL_ERR;
return NNACL_ERR;
}
tmp_avg = tmp_avg / (float)real_count;
tmp_avg = fmax(tmp_avg, minf);


+ 1
- 1
mindspore/lite/nnacl/fp32/pooling.h View File

@@ -28,7 +28,7 @@
extern "C" {
#endif
int AvgPooling(const float *input_ptr, float *output_ptr, PoolingParameter *pooling_param, int task_id, float minf,
float maxf);
float maxf);
void MaxPooling(const float *input_ptr, float *output_ptr, PoolingParameter *pooling_param, int task_id, float minf,
float maxf);
#ifdef __cplusplus


Loading…
Cancel
Save