| 28 | namespace tflite { |
| 29 | |
| 30 | inline void GetActivationMinMax(FusedActivationFunctionType ac, |
| 31 | float* output_activation_min, |
| 32 | float* output_activation_max) { |
| 33 | switch (ac) { |
| 34 | case FusedActivationFunctionType::kNone: |
| 35 | *output_activation_min = std::numeric_limits<float>::lowest(); |
| 36 | *output_activation_max = std::numeric_limits<float>::max(); |
| 37 | break; |
| 38 | case FusedActivationFunctionType::kRelu: |
| 39 | *output_activation_min = 0.f; |
| 40 | *output_activation_max = std::numeric_limits<float>::max(); |
| 41 | break; |
| 42 | case FusedActivationFunctionType::kRelu1: |
| 43 | *output_activation_min = -1.f; |
| 44 | *output_activation_max = 1.f; |
| 45 | break; |
| 46 | case FusedActivationFunctionType::kRelu6: |
| 47 | *output_activation_min = 0.f; |
| 48 | *output_activation_max = 6.f; |
| 49 | break; |
| 50 | } |
| 51 | } |
| 52 | |
| 53 | inline float ActivationFunctionWithMinMax(float x, float output_activation_min, |
| 54 | float output_activation_max) { |