burn-jit 0.16.1

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 conv2d;
mod conv3d;
mod conv_transpose3d;
mod deform_conv2d;
mod deform_conv_transpose2d;
mod error;

pub(crate) use conv2d::*;
pub(crate) use conv3d::*;
pub(crate) use conv_transpose3d::*;
pub(crate) use deform_conv2d::*;
pub(crate) use deform_conv_transpose2d::*;
pub(crate) use error::*;

pub use conv2d::{conv2d, conv_transpose2d, nchw_to_nhwc, Conv2dStrategy, ConvTranspose2dStrategy};