Skip to main content

rustorch_core/backend/
mod.rs

1#[derive(Debug, Clone, Copy, PartialEq, Eq)]
2pub enum Activation {
3    None = 0,
4    ReLU = 1,
5    Sigmoid = 2,
6    Tanh = 3,
7}
8
9#[cfg(feature = "wgpu_backend")]
10pub mod wgpu;