diff options
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())); } } |