xref: /aosp_15_r20/external/armnn/src/backends/reference/workloads/Activation.hpp (revision 89c4ff92f2867872bb9e2354d150bf0c8c502810)
1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #include "BaseIterator.hpp"
7 
8 #include <armnn/Tensor.hpp>
9 #include <armnn/Types.hpp>
10 
11 namespace armnn
12 {
13 float Activation(float in,
14                  ActivationFunction function,
15                  float a,
16                  float b);
17 
18 void Activation(Decoder<float>& in,
19                 Encoder<float>& out,
20                 const TensorInfo& tensorInfo,
21                 ActivationFunction function,
22                 float a,
23                 float b);
24 
25 } //namespace armnn
26