You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Extend to 1D and 3D the torch to linalg lowering of the average pool operator with count_include_pad = false (#4035)
Currently the avg_pool2d PyTorch operation supports the cases where
count_include_pad is true and false, but the avg_pool1d and avg_pool3d
only the true case is supported (which is simpler).
The count_include_pad = false support for avg_pool2d was added by
@AmosLewis in this change (reviewed by @rsuderman and @nirvedhmeshram) :
#3235
In this change I generalized the logic added above. I also did some
refactoring to the original code to reduce the size of the functions and
to avoid redundancy when possible.
@sahas3@dixinzhou@rafaelubal
---------
Co-authored-by: Ivan Garcia <[email protected]>
0 commit comments