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

    impl BoolVisionOps for LibTorch {}
    impl IntVisionOps for LibTorch {}
    impl FloatVisionOps for LibTorch {}
    impl VisionBackend for LibTorch {}
}

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

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