diff options
Diffstat (limited to 'src/backends/reference/workloads/Mean.hpp')
-rw-r--r-- | src/backends/reference/workloads/Mean.hpp | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/backends/reference/workloads/Mean.hpp b/src/backends/reference/workloads/Mean.hpp index 38c2e39653..dfb0302bf9 100644 --- a/src/backends/reference/workloads/Mean.hpp +++ b/src/backends/reference/workloads/Mean.hpp @@ -7,6 +7,7 @@ #include "armnn/DescriptorsFwd.hpp" #include "armnn/Tensor.hpp" +#include "BaseIterator.hpp" #include <vector> @@ -15,7 +16,7 @@ namespace armnn void Mean(const TensorInfo& inputInfo, const TensorInfo& outputInfo, const std::vector<unsigned int>& axis, - const float* inputData, - float* outputData); + Decoder<float>& input, + Encoder<float>& output); } //namespace armnn |