Pooling code does not allow sliding window starting in right padded region #46929
Labels
module: pooling
triaged
This issue has been looked at a team member, and triaged and prioritized into an appropriate module
The current pooling code does not allow the sliding window to start inside the right padding when ceil_mode=True which is inconsistent with the equation on the docs. This PR #46558 adds a note to the documentation but does not address this issue because it requires further investigation into existing code assumptions.
The code that computes the output shape here does not include right padding on ceil_mode correction.
pytorch/aten/src/ATen/native/Pool.h
Line 34 in bbe5bfa
cc @heitorschueroff
The text was updated successfully, but these errors were encountered: