burn-cubecl 0.23.0-pre.1

Generic backend that can be compiled just-in-time to any shader language target
Documentation
1
2
3
4
5
6
7
mod avg_pool3d;
mod base;
mod max_pool3d;

pub(crate) use avg_pool3d::*;
pub(crate) use base::*;
pub(crate) use max_pool3d::*;