burn-mpsgraph 0.0.1

Apple MPSGraph backend for the Burn deep learning framework
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18

pub mod activation;
pub mod bool_tensor;
pub mod float_tensor;
pub mod int_tensor;
pub mod module;
pub mod qtensor;
pub mod transaction;

/// Get a seeded StdRng (clone from global seed), or a fresh one from thread_rng.
pub(crate) fn get_seeded_rng() -> rand::rngs::StdRng {
    use rand::SeedableRng;
    let lock = crate::backend::SEED.lock().unwrap();
    if let Some(ref rng) = *lock {
        rng.clone()
    } else {
        rand::rngs::StdRng::from_rng(rand::thread_rng()).unwrap()
    }
}