burn-backend 0.22.0

Core backend interfaces and data structures for executing tensor operations in Burn.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

mod base;
mod device;
mod memory_pools;
mod primitive;
mod profile;
mod router_device_type;

pub use base::*;
pub use device::*;
pub use memory_pools::*;
pub use primitive::*;
pub use profile::*;
pub use router_device_type::*;

/// Backend operations on tensors.
pub mod ops;

/// Distributed backend extension.
pub mod distributed;