diff options
| author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2019-06-06 11:12:32 +0100 |
|---|---|---|
| committer | Teresa Charlin <teresa.charlinreyes@arm.com> | 2019-06-06 11:12:32 +0100 |
| commit | a3b20473ea0583ba4c6a2c17ba2d59ed958c0aa7 (patch) | |
| tree | 435c83dbbaadcd45df98359ee1d762daf2444994 /src/backends/reference/workloads/Pooling2d.hpp | |
| parent | 6331f91a4a1cb1ad16c569d98bb9ddf704788464 (diff) | |
| download | armnn-a3b20473ea0583ba4c6a2c17ba2d59ed958c0aa7.tar.gz | |
IVGCVSW-3142 Refactor reference Pooling2d workload
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I94c973ab747309c0214268c9c39f6d8f3fc7b255
Diffstat (limited to 'src/backends/reference/workloads/Pooling2d.hpp')
| -rw-r--r-- | src/backends/reference/workloads/Pooling2d.hpp | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/src/backends/reference/workloads/Pooling2d.hpp b/src/backends/reference/workloads/Pooling2d.hpp index da56b25c4e..182f9bd262 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 |