docs/core_2activation_8hpp_source.html

 #pragma once
 #include <dlprim/tensor.hpp>
 #include <dlprim/context.hpp>
 namespace dlprim {
 namespace core {

     void activation_forward(Tensor &x,Tensor &y,StandardActivations activation, ExecutionContext const &ec);

     void activation_backward(Tensor &dx,Tensor &dy,Tensor &y,StandardActivations activation, float factor, ExecutionContext const &ec);

 } // core
 } // dlprim
dlprim::core::activation_forward
void activation_forward(Tensor &x, Tensor &y, StandardActivations activation, ExecutionContext const &ec)
Appliy activation on X save to Y, Y can be same as X.

dlprim::core::activation_backward
void activation_backward(Tensor &dx, Tensor &dy, Tensor &y, StandardActivations activation, float factor, ExecutionContext const &ec)
Backward aclivation computed dx = dx * factor + backward(y,dy)

dlprim
Mane namespace.
Definition: context.hpp:9

dlprim::StandardActivations
StandardActivations
Parameterless Activations that can be embedded to general kernels like inner product or convolution...
Definition: definitions.hpp:266