Skip to main content

burn_vision/backends/cpu/
ops.rs

1#[cfg(feature = "tch")]
2mod tch {
3    use crate::{BoolVisionOps, FloatVisionOps, IntVisionOps, QVisionOps, VisionBackend};
4    use burn_tch::{LibTorch, TchElement};
5
6    impl<E: TchElement> BoolVisionOps for LibTorch<E> {}
7    impl<E: TchElement> IntVisionOps for LibTorch<E> {}
8    impl<E: TchElement> FloatVisionOps for LibTorch<E> {}
9    impl<E: TchElement> QVisionOps for LibTorch<E> {}
10    impl<E: TchElement> VisionBackend for LibTorch<E> {}
11}
12
13#[cfg(feature = "flex")]
14mod flex {
15    use crate::{BoolVisionOps, FloatVisionOps, IntVisionOps, QVisionOps, VisionBackend};
16    use burn_flex::Flex;
17
18    impl BoolVisionOps for Flex {}
19    impl IntVisionOps for Flex {}
20    impl FloatVisionOps for Flex {}
21    impl QVisionOps for Flex {}
22    impl VisionBackend for Flex {}
23}