rstsr-core 0.7.3

An n-Dimension Rust Tensor Toolkit
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109

use crate::prelude_dev::*;

/* #region op_func */

#[allow(non_camel_case_types)]
#[allow(clippy::too_many_arguments)]
pub trait Op_MutC_RefA_RefB_API<TA, TB, TC, D, F>
where
    D: DimAPI,
    F: FnMut(&mut MaybeUninit<TC>, &TA, &TB) + ?Sized,
    Self: DeviceAPI<TA> + DeviceAPI<TB> + DeviceAPI<MaybeUninit<TC>>,
{
    fn op_mutc_refa_refb_func(
        &self,
        c: &mut <Self as DeviceRawAPI<MaybeUninit<TC>>>::Raw,
        lc: &Layout<D>,
        a: &<Self as DeviceRawAPI<TA>>::Raw,
        la: &Layout<D>,
        b: &<Self as DeviceRawAPI<TB>>::Raw,
        lb: &Layout<D>,
        f: &mut F,
    ) -> Result<()>;
}

#[allow(non_camel_case_types)]
pub trait Op_MutC_RefA_NumB_API<TA, TB, TC, D, F>
where
    D: DimAPI,
    F: FnMut(&mut MaybeUninit<TC>, &TA, &TB) + ?Sized,
    Self: DeviceAPI<TA> + DeviceAPI<MaybeUninit<TC>>,
{
    fn op_mutc_refa_numb_func(
        &self,
        c: &mut <Self as DeviceRawAPI<MaybeUninit<TC>>>::Raw,
        lc: &Layout<D>,
        a: &<Self as DeviceRawAPI<TA>>::Raw,
        la: &Layout<D>,
        b: TB,
        f: &mut F,
    ) -> Result<()>;
}

#[allow(non_camel_case_types)]
pub trait Op_MutC_NumA_RefB_API<TA, TB, TC, D, F>
where
    D: DimAPI,
    F: FnMut(&mut MaybeUninit<TC>, &TA, &TB) + ?Sized,
    Self: DeviceAPI<TB> + DeviceAPI<MaybeUninit<TC>>,
{
    fn op_mutc_numa_refb_func(
        &self,
        c: &mut <Self as DeviceRawAPI<MaybeUninit<TC>>>::Raw,
        lc: &Layout<D>,
        a: TA,
        b: &<Self as DeviceRawAPI<TB>>::Raw,
        lb: &Layout<D>,
        f: &mut F,
    ) -> Result<()>;
}

#[allow(non_camel_case_types)]
pub trait Op_MutA_RefB_API<TA, TB, D, F>
where
    D: DimAPI,
    F: FnMut(&mut MaybeUninit<TA>, &TB) + ?Sized,
    Self: DeviceAPI<MaybeUninit<TA>> + DeviceAPI<TB>,
{
    fn op_muta_refb_func(
        &self,
        a: &mut <Self as DeviceRawAPI<MaybeUninit<TA>>>::Raw,
        la: &Layout<D>,
        b: &<Self as DeviceRawAPI<TB>>::Raw,
        lb: &Layout<D>,
        f: &mut F,
    ) -> Result<()>;
}

#[allow(non_camel_case_types)]
pub trait Op_MutA_NumB_API<TA, TB, D, F>
where
    D: DimAPI,
    F: FnMut(&mut MaybeUninit<TA>, &TB) + ?Sized,
    Self: DeviceAPI<MaybeUninit<TA>>,
{
    fn op_muta_numb_func(
        &self,
        a: &mut <Self as DeviceRawAPI<MaybeUninit<TA>>>::Raw,
        la: &Layout<D>,
        b: TB,
        f: &mut F,
    ) -> Result<()>;
}

#[allow(non_camel_case_types)]
pub trait Op_MutA_API<T, D, F>
where
    D: DimAPI,
    F: FnMut(&mut MaybeUninit<T>) + ?Sized,
    Self: DeviceAPI<MaybeUninit<T>>,
{
    fn op_muta_func(
        &self,
        a: &mut <Self as DeviceRawAPI<MaybeUninit<T>>>::Raw,
        la: &Layout<D>,
        f: &mut F,
    ) -> Result<()>;
}

/* #endregion */