diff options
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 |