ruDNN 0.1.40

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

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

pub mod pool2d;

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