burn-vision 0.21.0

Vision processing operations for burn tensors
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

#[cfg(feature = "tch")]
mod tch {
    use crate::{BoolVisionOps, FloatVisionOps, IntVisionOps, QVisionOps, VisionBackend};
    use burn_tch::{LibTorch, TchElement};

    impl<E: TchElement> BoolVisionOps for LibTorch<E> {}
    impl<E: TchElement> IntVisionOps for LibTorch<E> {}
    impl<E: TchElement> FloatVisionOps for LibTorch<E> {}
    impl<E: TchElement> QVisionOps for LibTorch<E> {}
    impl<E: TchElement> VisionBackend for LibTorch<E> {}
}

#[cfg(feature = "flex")]
mod flex {
    use crate::{BoolVisionOps, FloatVisionOps, IntVisionOps, QVisionOps, VisionBackend};
    use burn_flex::Flex;

    impl BoolVisionOps for Flex {}
    impl IntVisionOps for Flex {}
    impl FloatVisionOps for Flex {}
    impl QVisionOps for Flex {}
    impl VisionBackend for Flex {}
}