fix train op acc

This commit is contained in:
yefeng 2021-08-12 15:29:16 +08:00
parent a5388a5227
commit 746d6654fc
1 changed files with 1 additions and 1 deletions

View File

@ -170,7 +170,7 @@ void MaxPoolingGrad(const float *input_ptr, const float *dy_ptr, float *output_p
float delta = dyPtr[idx];
for (int kh = kh_s; kh < kh_e; kh++) {
int xh = yh * stride_h + kh - pad_h;
for (int kw = kw_e; kw < kw_s; kw++) {
for (int kw = kw_s; kw < kw_e; kw++) {
int xw = yw * stride_w + kw - pad_w;
int val_idx = (xw + in_w * xh) * channel + ic;
float val = inPtr[val_idx];