candela-tensor 0.2.0

A lazy, graph-based tensor engine in Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33

#[macro_export]
macro_rules! branch_fast_iter {
    ($value:expr => $name:ident, $body:expr) => {
        match $value {
            $crate::tensor::storage::IterImpl::Contiguous($name) => $body,
            $crate::tensor::storage::IterImpl::NotContiguous($name) => $body,
        }
    };
}

#[macro_export]
macro_rules! branch_duo_fast_iter {
    ($value1:expr => $name1:ident, $value2:expr => $name2:ident, $body:expr) => {
        match ($value1, $value2) {
            (
                $crate::tensor::storage::IterImpl::Contiguous($name1),
                $crate::tensor::storage::IterImpl::Contiguous($name2),
            ) => $body,
            (
                $crate::tensor::storage::IterImpl::NotContiguous($name1),
                $crate::tensor::storage::IterImpl::Contiguous($name2),
            ) => $body,
            (
                $crate::tensor::storage::IterImpl::Contiguous($name1),
                $crate::tensor::storage::IterImpl::NotContiguous($name2),
            ) => $body,
            (
                $crate::tensor::storage::IterImpl::NotContiguous($name1),
                $crate::tensor::storage::IterImpl::NotContiguous($name2),
            ) => $body,
        }
    };
}