burn_jit/kernel/pool/
mod.rs

1mod adaptive_avg_pool2d;
2mod adaptive_avg_pool2d_backward;
3mod avg_pool2d;
4mod avg_pool2d_backward;
5mod max_pool2d;
6mod max_pool2d_backward;
7
8pub(super) mod pool2d;
9
10pub(crate) use adaptive_avg_pool2d::*;
11pub(crate) use adaptive_avg_pool2d_backward::*;
12pub(crate) use avg_pool2d::*;
13pub(crate) use avg_pool2d_backward::*;
14pub(crate) use max_pool2d::*;
15pub(crate) use max_pool2d_backward::*;