diff options
Diffstat (limited to 'src/backends/reference/workloads/Rsqrt.hpp')
-rw-r--r-- | src/backends/reference/workloads/Rsqrt.hpp | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/backends/reference/workloads/Rsqrt.hpp b/src/backends/reference/workloads/Rsqrt.hpp index 35cacede66..ffc6b18d13 100644 --- a/src/backends/reference/workloads/Rsqrt.hpp +++ b/src/backends/reference/workloads/Rsqrt.hpp @@ -3,6 +3,7 @@ // SPDX-License-Identifier: MIT // +#include "BaseIterator.hpp" #include <armnn/Tensor.hpp> #include <armnn/Types.hpp> @@ -11,8 +12,8 @@ namespace armnn /// Performs the reciprocal squareroot function elementwise /// on the inputs to give the outputs. -void Rsqrt(const float* in, - float* out, +void Rsqrt(Decoder<float>& in, + Encoder<float>& out, const TensorInfo& tensorInfo); } //namespace armnn |