Skip to main content

burn_candle/ops/
activation.rs

1use burn_backend::{ops::ActivationOps, tensor::FloatTensor};
2
3use crate::{
4    Candle, CandleTensor,
5    element::{CandleElement, FloatCandleElement, IntCandleElement},
6    tensor,
7};
8
9impl<F: FloatCandleElement, I: IntCandleElement> ActivationOps<Self> for Candle<F, I> {
10    fn gelu(tensor: FloatTensor<Self>) -> FloatTensor<Self> {
11        CandleTensor::new(tensor.tensor.gelu().unwrap())
12    }
13
14    fn relu(tensor: FloatTensor<Self>) -> FloatTensor<Self> {
15        CandleTensor::new(tensor.tensor.relu().unwrap())
16    }
17}