Support ragged patch grids in attentional_pool - #195

Merged

amazloumi merged 8 commits into

mainfrom

feat/ragged-attentional-pool

Sep 8, 2026

Commits on Sep 2, 2026

Support ragged grids in attentional_pool
amazloumi
committed
86b253b
update the error message
amazloumi
committed
6cd5015
Refactor pooling adapters to share a _pad_grid_to_windows helper for ragged-edge handling
amazloumi
committed
43363bf
Reword stale require_divisible test comment to reflect ragged-capable pooling
amazloumi
committed
15cd5bf
Remove unused divisible-only pooling seam; add ragged-pool corner-case tests
amazloumi
committed
ab36ed1
Mask avgpool's numerator through `valid`, not the pad value
amazloumi
committed
8f1fb52
Add ragged-pool corner cases and an independent per-window reference
amazloumi
committed
5670599
Document ragged pooling and the pooled_token_count signature change
amazloumi
committed
e997c27