Skip to content
Snippets Groups Projects
Commit 1b92635b authored by Olivier BICHLER's avatar Olivier BICHLER
Browse files

Revert "Added CountIncludePad attribute"

This reverts commit 79e3c00f
parent 79e3c00f
No related branches found
No related tags found
2 merge requests!73version 0.2.3,!70Multiple refactors
Pipeline #49216 failed
...@@ -91,14 +91,8 @@ void AvgPoolingImpl2D_cpu_forward_kernel(const AvgPooling_Op<2>::Attrs &attrs, ...@@ -91,14 +91,8 @@ void AvgPoolingImpl2D_cpu_forward_kernel(const AvgPooling_Op<2>::Attrs &attrs,
output[oIndexFull] += input[iIndex + (ix+sx)*dims[3] + (iy+sy)]; output[oIndexFull] += input[iIndex + (ix+sx)*dims[3] + (iy+sy)];
} }
} }
if (std::get<2>(attrs)) { // padding not used
// count_include_pad is true output[oIndexFull] /= (sxMax - sxMin) * (syMax - syMin);
output[oIndexFull] /= std::get<1>(attrs)[0] * std::get<1>(attrs)[1];
}
else {
// padding not used
output[oIndexFull] /= (sxMax - sxMin) * (syMax - syMin);
}
} }
} }
} }
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment