commit a3b20473ea0583ba4c6a2c17ba2d59ed958c0aa7
author Teresa Charlin <teresa.charlinreyes@arm.com> Thu Jun 06 11:12:32 2019 +0100
committer Teresa Charlin <teresa.charlinreyes@arm.com> Thu Jun 06 11:12:32 2019 +0100
tree 435c83dbbaadcd45df98359ee1d762daf2444994
parent 6331f91a4a1cb1ad16c569d98bb9ddf704788464

IVGCVSW-3142 Refactor reference Pooling2d workload

Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I94c973ab747309c0214268c9c39f6d8f3fc7b255

src/backends/reference/workloads/Pooling2d.hpp

diff --git a/src/backends/reference/workloads/Pooling2d.hpp b/src/backends/reference/workloads/Pooling2d.hpp
index da56b25..182f9bd 100644
--- a/src/backends/reference/workloads/Pooling2d.hpp
+++ b/src/backends/reference/workloads/Pooling2d.hpp
@@ -8,14 +8,14 @@
 #include <armnn/Descriptors.hpp>
 #include <armnn/Tensor.hpp>
 
+#include "BaseIterator.hpp"
+
 namespace armnn
 {
-
 /// Computes the Pooling2d operation.
-void Pooling2d(const float* in,
-               float* out,
+void Pooling2d(Decoder<float>& rInputDecoder,
+               Encoder<float>& rOutputEncoder,
                const TensorInfo& inputInfo,
                const TensorInfo& outputInfo,
                const Pooling2dDescriptor& params);
-
 } //namespace armnn