flax
Adding "count_include_pad" argument to flax.linen.pooling.avg_pool
#2451
Merged
Merged
Adding "count_include_pad" argument to flax.linen.pooling.avg_pool #2451
copybara-service
merged
5 commits into
google:main
from
dslisleedh:avg_pool_countpad
Add count_include_pad argument to flax.linen.pooling.avg_pool
9f5ad668
Add count_include_pad argument to flax.linen.pooling.avg_pool
ff97879c
cgarciae
approved these changes
on 2022-09-14
Add count_include_pad argument to flax.linen.pooling.avg_pool / Fix d…
8af16050
Add count_include_pad argument to flax.linen.pooling.avg_pool / Fix d…
c73a4b79
Add test_avg_pool_no_batch under test.linen.linen_test.PoolTest
d4fa16d3
cgarciae
approved these changes
on 2022-09-21
Login to write a write a comment.
Login via GitHub