class ov::op::util::ActivationFunction¶
Overview¶
Class representing activation function used in RNN cells. More…
#include <activation_functions.hpp>
class ActivationFunction
{
public:
// construction
ActivationFunction(ActivationFunctionType f, float alpha, float beta);
ActivationFunction(ActivationFunctionType f, float alpha);
ActivationFunction(ActivationFunctionType f);
ActivationFunction();
// methods
std::shared_ptr<Node> operator () (const std::shared_ptr<Node>& arg) const;
void set_alpha(float alpha);
void set_beta(float beta);
};