nnfw::cker::ActivationFunctor Class Reference

#include <PortableTensorUtils.h>

Public Member Functions
	ActivationFunctor (FusedActivationFunctionType act)
float	operator() (float a) const

Detailed Description

Definition at line 33 of file PortableTensorUtils.h.

Constructor & Destructor Documentation

ActivationFunctor()

nnfw::cker::ActivationFunctor::ActivationFunctor ( FusedActivationFunctionType  act )

inlineexplicit

Definition at line 36 of file PortableTensorUtils.h.

: act_(act) {}

Member Function Documentation

operator()()

float nnfw::cker::ActivationFunctor::operator() ( float  a ) const

inline

Definition at line 38 of file PortableTensorUtils.h.

  {
    switch (act_)
    {
      case FusedActivationFunctionType::kNone:
        return a;
      case FusedActivationFunctionType::kRelu:
        return a < 0.f ? 0.f : a;
      case FusedActivationFunctionType::kRelu6:
        return std::max(0.f, std::min(a, 6.f));
      case FusedActivationFunctionType::kTanh:
        return std::tanh(a);
      case FusedActivationFunctionType::kSigmoid:
        return 1.0f / (1.0f + std::exp(-a));
      default:
        // TODO(aselle): More informative fatal error!
        exit(1);
    }
  }

References nnfw::cker::kNone, nnfw::cker::kRelu, nnfw::cker::kRelu6, nnfw::cker::kSigmoid, and nnfw::cker::kTanh.

---

The documentation for this class was generated from the following file:

• compute/cker/include/cker/PortableTensorUtils.h