singe-kernel 0.1.0-alpha.4

Reusable CPU and GPU kernels.
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

#[cfg(feature = "kernel-activation")]
#[doc(hidden)]
#[path = "../../../../-generated/cutile/f64/activation.rs"]
pub mod activation;
#[cfg(feature = "kernel-cast")]
#[doc(hidden)]
#[path = "../../../../-generated/cutile/f64/cast.rs"]
pub mod cast;
#[cfg(feature = "kernel-elementwise")]
#[doc(hidden)]
#[path = "../../../../-generated/cutile/f64/elementwise.rs"]
pub mod elementwise;
#[cfg(feature = "kernel-quantization")]
#[doc(hidden)]
#[path = "../../../../-generated/cutile/f64/quantization.rs"]
pub mod quantization;
#[cfg(feature = "kernel-reduction")]
#[doc(hidden)]
#[path = "../../../../-generated/cutile/f64/reduction.rs"]
pub mod reduction;
#[cfg(feature = "kernel-scalar")]
#[doc(hidden)]
#[path = "../../../../-generated/cutile/f64/scalar.rs"]
pub mod scalar;
#[doc(hidden)]
#[path = "../../../../-generated/cutile/f64/softmax.rs"]
pub mod softmax;
#[cfg(feature = "kernel-unary")]
#[doc(hidden)]
#[path = "../../../../-generated/cutile/f64/unary.rs"]
pub mod unary;

#[cfg(feature = "kernel-embedding")]
pub mod embedding;
#[cfg(feature = "kernel-gather")]
pub mod gather;
#[cfg(feature = "kernel-normalization")]
pub mod normalization;
#[cfg(feature = "kernel-pooling")]
pub mod pooling;
#[cfg(feature = "kernel-positional")]
pub mod positional;
#[cfg(feature = "kernel-shape")]
pub mod shape;
#[cfg(feature = "kernel-softmax")]
#[cfg(any(
    feature = "kernel-activation",
    feature = "kernel-cast",
    feature = "kernel-elementwise",
    feature = "kernel-quantization",
    feature = "kernel-reduction",
    feature = "kernel-scalar",
    feature = "kernel-softmax",
    feature = "kernel-unary",
))]
#[doc(hidden)]
pub mod utility;