diff options
Diffstat (limited to 'candle-nn/src/activation.rs')
| -rw-r--r-- | candle-nn/src/activation.rs | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/candle-nn/src/activation.rs b/candle-nn/src/activation.rs index 9554e68a..0db3edc9 100644 --- a/candle-nn/src/activation.rs +++ b/candle-nn/src/activation.rs @@ -7,8 +7,8 @@ pub enum Activation { Elu(f64), } -impl Activation { - pub fn forward(&self, xs: &Tensor) -> candle::Result<Tensor> { +impl super::Module for Activation { + fn forward(&self, xs: &Tensor) -> candle::Result<Tensor> { match self { Self::Gelu => xs.gelu(), Self::Relu => xs.relu(), |