aboutsummaryrefslogtreecommitdiff
path: root/src/backends/reference/workloads/RefFloorWorkload.cpp
diff options
context:
space:
mode:
authorJames Conroy <james.conroy@arm.com>2019-05-30 16:36:59 +0100
committerJames Conroy <james.conroy@arm.com>2019-05-31 15:56:38 +0100
commit83735b177a4c71952d36e7a7919537b5a4d6d49e (patch)
tree51a2e30b1f6caeea9994848c3be6f93471d03b64 /src/backends/reference/workloads/RefFloorWorkload.cpp
parent983daec95bd3b3e5558771e995092981f2df2373 (diff)
downloadarmnn-83735b177a4c71952d36e7a7919537b5a4d6d49e.tar.gz
IVGCVSW-3177 Refactor Floor reference workload
* Renamed RefFloorFloat32Workload to RefFloorWorkload and updated references to reflect this change. * RefFloorWorkload now uses Decoders/Encoders and supports the use of multiple data types. * Deleted FloorTestImpl.hpp and moved its contents into LayerTests.hpp. Change-Id: Ie079d05f2f6a578172f0fe3024f9607c030bce64 Signed-off-by: James Conroy <james.conroy@arm.com>
Diffstat (limited to 'src/backends/reference/workloads/RefFloorWorkload.cpp')
-rw-r--r--src/backends/reference/workloads/RefFloorWorkload.cpp38
1 files changed, 38 insertions, 0 deletions
diff --git a/src/backends/reference/workloads/RefFloorWorkload.cpp b/src/backends/reference/workloads/RefFloorWorkload.cpp
new file mode 100644
index 0000000000..d96b18b530
--- /dev/null
+++ b/src/backends/reference/workloads/RefFloorWorkload.cpp
@@ -0,0 +1,38 @@
+//
+// Copyright © 2017 Arm Ltd. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#include "RefFloorWorkload.hpp"
+
+#include "Decoders.hpp"
+#include "Encoders.hpp"
+#include "RefWorkloadUtils.hpp"
+#include "Profiling.hpp"
+
+namespace armnn
+{
+
+void RefFloorWorkload::Execute() const
+{
+ ARMNN_SCOPED_PROFILING_EVENT(Compute::CpuRef, "RefFloorFloat32Workload_Execute");
+
+ const TensorInfo &inputTensorInfo = GetTensorInfo(m_Data.m_Inputs[0]);
+ std::unique_ptr<Decoder<float>> decoderPtr = MakeDecoder<float>(inputTensorInfo, m_Data.m_Inputs[0]->Map());
+ Decoder<float> &decoder = *decoderPtr;
+
+ const TensorInfo &outputTensorInfo = GetTensorInfo(m_Data.m_Outputs[0]);
+ std::unique_ptr<Encoder<float>> encoderPtr = MakeEncoder<float>(outputTensorInfo, m_Data.m_Outputs[0]->Map());
+ Encoder<float> &encoder = *encoderPtr;
+
+ unsigned int numElements = GetTensorInfo(m_Data.m_Inputs[0]).GetNumElements();
+
+ for (unsigned int i = 0; i < numElements; ++i)
+ {
+ encoder.Set(floorf(decoder.Get()));
+ ++decoder;
+ ++encoder;
+ }
+}
+
+} //namespace armnn