diff options
author | nikraj01 <nikhil.raj@arm.com> | 2019-06-06 10:31:27 +0100 |
---|---|---|
committer | Nikhil Raj Arm <nikhil.raj@arm.com> | 2019-06-06 16:06:10 +0000 |
commit | 99a663140294afd2a4ea91ccc61b7266f735b46a (patch) | |
tree | bf6984e3734f0d5182a7fa510b80e3600e640e4f /src/backends/reference/workloads/Rsqrt.hpp | |
parent | 0434df6030eee9bb6842b02a8b28598cfe8f3460 (diff) | |
download | armnn-99a663140294afd2a4ea91ccc61b7266f735b46a.tar.gz |
IVGCVSW-3211 Refactor reference Rsqrt workload
Change-Id: Ia413c6b5352dbb3390e7d84e837a542c24ae8813
Signed-off-by: nikraj01 <nikhil.raj@arm.com>
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 |