burn-cubecl 0.21.0-pre.4

Generic backend that can be compiled just-in-time to any shader language target
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15

mod adaptive_avg_pool2d;
mod adaptive_avg_pool2d_backward;
mod avg_pool2d;
mod avg_pool2d_backward;
mod max_pool2d;
mod max_pool2d_backward;

pub(super) mod pool2d;

pub(crate) use adaptive_avg_pool2d::*;
pub(crate) use adaptive_avg_pool2d_backward::*;
pub(crate) use avg_pool2d::*;
pub(crate) use avg_pool2d_backward::*;
pub(crate) use max_pool2d::*;
pub(crate) use max_pool2d_backward::*;