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.cpp | |
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.cpp')
-rw-r--r-- | src/backends/reference/workloads/Rsqrt.cpp | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/src/backends/reference/workloads/Rsqrt.cpp b/src/backends/reference/workloads/Rsqrt.cpp index cee38fc1f1..5abc2c8f7b 100644 --- a/src/backends/reference/workloads/Rsqrt.cpp +++ b/src/backends/reference/workloads/Rsqrt.cpp @@ -10,13 +10,15 @@ namespace armnn { -void Rsqrt(const float* in, - float* out, +void Rsqrt(Decoder<float>& in, + Encoder<float>& out, const TensorInfo& tensorInfo) { - for (size_t i = 0; i < tensorInfo.GetNumElements(); i++) + for (unsigned int i = 0; i < tensorInfo.GetNumElements(); ++i) { - out[i] = 1.f / sqrtf(in[i]); + out[i]; + in[i]; + out.Set(1.f / sqrtf(in.Get())); } } |