From 33d46ff37e3272222a3c99641e1bbbdfe1e17d5d Mon Sep 17 00:00:00 2001 From: Antoni Olivier <olivier.antoni@cea.fr> Date: Wed, 5 Jun 2024 22:13:15 +0200 Subject: [PATCH] Add backward functions for ReLU, Sigmoid and Tanh --- .../aidge/backend/cpu/operator/ReLUImpl_backward_kernels.hpp | 2 +- .../aidge/backend/cpu/operator/SigmoidImpl_backward_kernels.hpp | 2 +- .../aidge/backend/cpu/operator/TanhImpl_backward_kernels.hpp | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/include/aidge/backend/cpu/operator/ReLUImpl_backward_kernels.hpp b/include/aidge/backend/cpu/operator/ReLUImpl_backward_kernels.hpp index 02b5e30b..e67a4588 100644 --- a/include/aidge/backend/cpu/operator/ReLUImpl_backward_kernels.hpp +++ b/include/aidge/backend/cpu/operator/ReLUImpl_backward_kernels.hpp @@ -23,7 +23,7 @@ void ReLUImpl_cpu_backward_kernel(const std::size_t inputLenght, const void* input_, const void* output_, const void* grad_output_, void* grad_input_) { const I* input = static_cast<const I*>(input_); - #const O* output = static_cast<const O*>(output_); + //const O* output = static_cast<const O*>(output_); const GO* grad_output = static_cast<const GO*>(grad_output_); GI* grad_input = static_cast<GI*>(grad_input_); for (std::size_t i = 0; i < inputLenght; ++i) { diff --git a/include/aidge/backend/cpu/operator/SigmoidImpl_backward_kernels.hpp b/include/aidge/backend/cpu/operator/SigmoidImpl_backward_kernels.hpp index e10f1d53..931a2e4d 100644 --- a/include/aidge/backend/cpu/operator/SigmoidImpl_backward_kernels.hpp +++ b/include/aidge/backend/cpu/operator/SigmoidImpl_backward_kernels.hpp @@ -22,7 +22,7 @@ template <class I, class O, class GI, class GO> void SigmoidImpl_cpu_backward_kernel(const std::size_t inputLenght, const void* input_, const void* output_, const void* grad_output_, void* grad_input_) { - #const I* input = static_cast<const I*>(input_); + //const I* input = static_cast<const I*>(input_); const O* output = static_cast<const O*>(output_); const GO* grad_output = static_cast<const GO*>(grad_output_); GI* grad_input = static_cast<GI*>(grad_input_); diff --git a/include/aidge/backend/cpu/operator/TanhImpl_backward_kernels.hpp b/include/aidge/backend/cpu/operator/TanhImpl_backward_kernels.hpp index 6780a4e3..3f49e4c4 100644 --- a/include/aidge/backend/cpu/operator/TanhImpl_backward_kernels.hpp +++ b/include/aidge/backend/cpu/operator/TanhImpl_backward_kernels.hpp @@ -22,7 +22,7 @@ template <class I, class O, class GI, class GO> void TanhImpl_cpu_backward_kernel(const std::size_t inputLenght, const void* input_, const void* output_, const void* grad_output_, void* grad_input_) { - #const I* input = static_cast<const I*>(input_); + //const I* input = static_cast<const I*>(input_); const O* output = static_cast<const O*>(output_); const GO* grad_output = static_cast<const GO*>(grad_output_); GI* grad_input = static_cast<GI*>(grad_input_); -- GitLab