ruDNN 0.1.71

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
24
25
26
27
28
29

mod backward_data;
mod backward_weight;
mod base;
mod conv_transpose2d;
mod conv_transpose3d;
mod deform_conv2d;
mod deform_conv_transpose2d;
mod direct;
mod forward;
mod im2col;

mod tune_key;

pub(crate) use {backward_data::*};
pub(crate) use {conv_transpose2d::*};
pub(crate) use {conv_transpose3d::*};
pub(crate) use {deform_conv_transpose2d::*};
pub(crate) use {deform_conv2d::*};
pub(crate) use {direct::*};
pub(crate) use {im2col::*};

pub use {base::*};
pub use {conv_transpose2d::ConvTranspose2dStrategy, conv_transpose2d::conv_transpose2d};

pub use tune_key::{ConvAutotuneKey, ConvTranspose2dAutotuneKey};

pub use conv_transpose3d::conv_transpose3d;
pub use deform_conv2d::deform_conv2d;
pub use deform_conv_transpose2d::deform_conv2d_backward;