ruDNN 0.1.82

Neural network operators for Ruda.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

mod adaptive_avg_pool2d;
mod adaptive_avg_pool2d_backward;
mod adaptive_avg_pool3d;
mod avg_pool2d;
mod avg_pool2d_backward;
mod avg_pool3d;
mod max_pool2d;
mod max_pool2d_aten;
mod max_pool2d_backward;
mod max_pool3d;

pub mod pool2d;

pub use adaptive_avg_pool2d::*;
pub use adaptive_avg_pool2d_backward::*;
pub use adaptive_avg_pool3d::*;
pub use avg_pool2d::*;
pub use avg_pool2d_backward::*;
pub use avg_pool3d::*;
pub use max_pool2d::*;
pub use max_pool2d_aten::*;
pub use max_pool2d_backward::*;
pub use max_pool3d::*;