burn/
backend.rs

1#[cfg(feature = "ndarray")]
2pub use burn_ndarray as ndarray;
3
4#[cfg(feature = "ndarray")]
5pub use ndarray::NdArray;
6
7#[cfg(feature = "autodiff")]
8pub use burn_autodiff as autodiff;
9
10#[cfg(feature = "remote")]
11pub use burn_remote as remote;
12#[cfg(feature = "remote")]
13pub use burn_remote::RemoteBackend;
14
15#[cfg(feature = "autodiff")]
16pub use burn_autodiff::Autodiff;
17
18#[cfg(feature = "wgpu")]
19pub use burn_wgpu as wgpu;
20
21#[cfg(feature = "wgpu")]
22pub use burn_wgpu::Wgpu;
23
24#[cfg(feature = "webgpu")]
25pub use burn_wgpu::WebGpu;
26
27#[cfg(feature = "vulkan")]
28pub use burn_wgpu::Vulkan;
29
30#[cfg(feature = "metal")]
31pub use burn_wgpu::Metal;
32
33#[cfg(feature = "cuda")]
34pub use burn_cuda as cuda;
35
36#[cfg(feature = "cuda")]
37pub use burn_cuda::Cuda;
38
39#[cfg(feature = "candle")]
40pub use burn_candle as candle;
41
42#[cfg(feature = "candle")]
43pub use burn_candle::Candle;
44
45#[cfg(feature = "rocm")]
46pub use burn_rocm as rocm;
47
48#[cfg(feature = "rocm")]
49pub use burn_rocm::Rocm;
50
51#[cfg(feature = "tch")]
52pub use burn_tch as libtorch;
53
54#[cfg(feature = "tch")]
55pub use burn_tch::LibTorch;
56
57#[cfg(feature = "router")]
58pub use burn_router::Router;
59
60#[cfg(feature = "router")]
61pub use burn_router as router;